ML Compiler Engineer

Nvidia • Seattle, WA

436d • $164,000 - $310,500

About The Position

NVIDIA is seeking outstanding ML compiler engineers to develop groundbreaking technologies in machine learning compilers and AI systems. The role involves innovating AI compiler solutions that integrate with NVIDIA's software stack to accelerate modern machine learning models, particularly large language models and other high-impact workloads. Engineers will collaborate closely with internal teams to push the latest developments in NVIDIA's products.

Requirements

• Bachelor's degree in Computer Science, Electrical Engineering, or related field (or equivalent experience); MS or PhD are preferred
• 7+ years of experience
• Strong experience in developing or using deep learning frameworks (e.g. PyTorch, JAX, TensorFlow, ONNX etc)
• Strong python and C/C++ programming skills

Nice To Haves

• Expertise in AI frameworks such as PyTorch, TensorFlow, and ONNX
• Expertise in machine compilers (e.g. Apache TVM, MLIR)
• Strong experience in GPU performance optimizations
• Strong experience in machine learning systems research and productionization
• Open source project ownership or contributions

Responsibilities

• Innovate and develop new machine learning compiler and systems technologies
• Design, implement, and optimize compilers for high impact AI workloads
• Co-design learning system solutions with current and future ML compiler and algorithm technologies
• Collaborate closely with other engineering teams at NVIDIA to build high impact solutions for machine learning acceleration

Benefits

• Equity
• Comprehensive benefits package

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Industry: Computer and Electronic Product Manufacturing
• Education Level: Bachelor's degree