Senior Software Engineer - Graph Compiler

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 optimizations

JOB 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.

REQUIRED KNOWLEDGE, SKILLS, AND ABILITIES

1. Strong experience in C++ programming
2. Hands on experience on Python
3. Strong understanding of algorithms and data structures.
4. Experience with debugging and profiling tools to analyze and optimize code performance.
5. Hands on experience in any of the ML graph compiler framework.
6. Knowledge of native parallel processing hardware architectures and optimization techniques is a plus

MANDATORY SKILLS

• Experience on C/C++, Python, LLVM, OpenCLC++, ML framework