ML Compiler Engineer

A company is looking for an ML Compiler Engineer to develop innovative technologies in machine learning compilers and AI systems.



Key Responsibilities

Innovate and develop new machine learning compiler and systems technologies
Design, implement, and optimize compilers for high impact AI workloads
Collaborate closely with other engineering teams to build solutions for machine learning acceleration


Required Qualifications

Bachelor's degree in Computer Science, Electrical Engineering, or related field; MS or PhD preferred
7+ years of experience in the field
Strong experience with deep learning frameworks (e.g., PyTorch, JAX, TensorFlow, ONNX)
Strong programming skills in Python and C/C++
Expertise in AI frameworks and machine compilers (e.g., Apache TVM, MLIR) is preferred