Machine Learning Compiler

About the Role

Lead a team of engineers focused on advancing machine learning compiler technologies for cutting-edge AI accelerators for the Cloud and Datacenter market. This role combines deep technical expertise with strategic leadership to deliver high-performance compiler solutions that enable next-generation AI workloads.

Responsibilities

• Technical Leadership: Drive innovation in ML compiler optimization algorithms, leveraging advanced mathematics, programming language theory, and modern compiler techniques. Oversee feature design, code reviews, and integration of compiler components into the AI software stack. Ensure robust testing, profiling, and performance tuning across diverse ML models and hardware targets.
• Team & Project Management: Manage and mentor a team of engineers, fostering technical growth and collaboration. Plan and execute projects to meet release deliverables, balancing tactical improvements with strategic roadmap goals. Coordinate cross-functional efforts with hardware, architecture, and AI framework teams.
• Strategic Vision: Shape the evolution of ML compiler technology within the organization, aligning with industry trends and internal product strategies.

Requirements

• Bachelor’s degree in Computer Science, Electrical Engineering, or related field and 10+ years of relevant experience; OR Master’s degree and 8+ years of experience; OR Ph.D. and 6+ years of experience in compiler development, ML systems, or related areas.

Preferred Skills & Experience

• Strong background in compiler design (LLVM, optimization passes, IR transformations).
• Expertise in ML frameworks (PyTorch, ONNX) and graph-level optimizations.
• Familiarity with ML hardware accelerators, memory hierarchies, and performance modeling.
• Solid C++ programming skills.
• Proven ability to lead teams and deliver complex software projects in fast-paced environments.

Why Join Us

You’ll work on state-of-the-art ML compilers, applying advanced algorithms to optimize AI workloads for next-generation hardware. If you’re passionate about pushing the boundaries of compiler technology and leading impactful teams, this is the role for you.

Pay

$160,600.00 - $240,800.00

Benefits

• Competitive annual discretionary bonus program and opportunity for annual RSU grants (employees on sales-incentive plans are not eligible for our annual bonus).
• Highly competitive benefits package designed to support your success at work, at home, and at play.