Senior Machine Learning Compiler Engineer

Job Summary

To develop groundbreaking technologies in machine learning compilers and AI systems, the full-time Senior Machine Learning Compiler Engineer will innovate and optimize compilers for high-impact AI workloads while collaborating closely with internal teams in a remote or onsite capacity.

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 engineering teams to build high-impact solutions for machine learning acceleration

Required qualifications

• Bachelor's degree in Computer Science, Electrical Engineering, or related field (MS or PhD preferred)
• 6+ years of experience in machine learning systems development, including ML compilers and LLM inference kernels
• Strong experience in developing or using deep learning frameworks (e.g., PyTorch, JAX)
• Proficient in Python and C/C++ programming