Senior Deep Learning Compiler Engineer - CUDA

Job Description

NVIDIA is looking for outstanding ML/DL compiler engineers to develop groundbreaking technologies in machine learning compilers and AI systems.

Requirements

• Bachelor's degree in Computer Science, Electrical Engineering, or related field (or equivalent experience); MS or PhD are preferred
• 5+ years (academic/ industry) experience with ML/DL systems development preferable for compilers
• Strong Python and C/C++ programming skills
• Expert experience in developing or using deep learning frameworks (e.g. PyTorch, JAX, Triton, etc.)
• Strong sense of ownership, fast learner, passion for quality and user experience

Responsibilities

• Design and implement Pythonic language interface for tile-aware GPU programming
• Optimizing compiler pipelines for efficient execution
• Integrate with AI/ML frameworks
• Develop performance critical primitives for tensor operations and memory operations
• Collaborate with hardware teams to co-design compiler optimizations for emerging GPU architectures, including Tensor Core utilization and distributed execution

Benefits

• Multiple relocation packages
• Two weeklong shutdowns (mid-summer and year-end) in the US (in addition to PTO)
• 8-week parental leave
• 9 Employee Resource Groups
• Annual bonus offering
• Flexible work arrangements
• Up to 6% 401K matching