Senior Software Engineer - Graph Compiler

Blaize

  • Hyderabad, Telangana
  • Permanent
  • Full-time
  • 2 months ago
Blaize provides a full-stack programmable processor architecture suite and low-code/no-code software platform that enables AI processing solutions for high-performance computing at the network's edge and in the data center. Blaize solutions deliver real-time insights and decision-making capabilities at low power consumption, high efficiency, minimal size and low cost. Blaize has raised over $330 million from strategic investors such as DENSO, Mercedes-Benz AG, Magna, and Samsung and financial investors such as Franklin Templeton, Temasek, GGV, Bess Ventures, BurTech LP LLC, Rizvi Traverse, and Ava Investors. Headquartered in El Dorado Hills (Calif.), Blaize has more than 200 employees worldwide with teams in San Jose (Calif.) and Cary (N.C.), and subsidiaries in Hyderabad (India), Leeds and Kings Langley (UK), and Abu Dhabi (UAE).JOB DESCRPTION:We are seeking a highly skilled ML Graph Compiler Engineer to join our team. The ideal candidate will have a strong background in C++ and experience in developing and optimizing graph compilers for machine learning models. You will work closely with our hardware and software teams to enhance the performance and efficiency of our ML frameworks, focusing on native parallel processing hardware and neural network optimizationsJOB RESPONSIBLITIES
  • Design, develop, and optimize graph compilers for machine learning models.
  • Implement compiler optimizations such as operator fusion, memory management, and code generation.
  • Develop and optimize kernels in Clang C++ (Clcpp) and assembly for neural networks.
  • Collaborate with hardware engineers to integrate and optimize code for native parallel processing hardware.
  • Analyze and improve the performance of ML models on Blaize hardware platforms.
  • Maintain and extend existing compiler infrastructure.
EDUCATION AND EXPERIENCE
  • A bachelor's/master's degree in computer science
  • 5+ years hands-on experience in software development
REQUIRED KNOWLEDGE, SKILLS, AND ABILITIES * Strong experience in C++ programming, including modern C++ standards (C++11/14/17).
  • Hands on experience on Python
  • Strong understanding of algorithms and data structures.
  • Experience with debugging and profiling tools to analyze and optimize code performance.
  • Hands on experience in any of the ML graph compiler framework.
  • Experience in developing and optimizing kernels in Clang C++ (Clcpp) and assembly is a plus
  • Knowledge of native parallel processing hardware architectures and optimization techniques is a plus
MANDATORY SKILLS
  • Experience on C/C++, Python, LLVM, OpenCLC++, ML framework
SOFT SKILLS
  • Strong analytical skills and ability to approach complex problems methodically
  • Ability to work independently and collaborate effectively with the teams
COMMUNICATION SKILLS
  • Strong written and verbal communication skills
  • Ability of explain complex technical concepts to non-technical stakeholders
Blaize is an equal opportunity employer. We pride ourselves on having a diverse workforce and we do not discriminate against any employee or applicant because of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition, or any other basis protected by law. We respect the gender, gender identity, and gender expression of our applicants and employees, and we honor requests for preferred pronouns. It is our policy to comply with all applicable national, state, and local laws pertaining to nondiscrimination and equal opportunity.

Blaize