ML Compiler Engineer

Nvidia • Seattle, WA

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

About The Position

The ML Compiler Engineer at NVIDIA will be responsible for developing innovative technologies in machine learning compilers and AI systems. This role involves creating AI compiler solutions that integrate with NVIDIA's software stack to enhance the performance of modern machine learning models. The engineer will work closely with internal teams to push the latest developments in NVIDIA's product offerings.

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 health benefits
• Paid time off
• Retirement savings plan
• Employee discounts

What This Job Offers

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