AI Compiler Engineer

Join NVIDIA's team at the forefront of AI compiler technology and help shape the future of accelerated computing.

As a key member of our team, you will contribute directly to the future of accelerated AI by designing, implementing, optimizing, and maintaining core compiler technologies that accelerate massive deep learning workloads.

Your role will involve:

• Leading the design, implementation, optimization, and maintenance of core compiler technologies
• Collaborating with engineers who built PyTorch for NVIDIA hardware
• Pioneering new features and staying at the forefront of framework development
• Analyzing performance and optimizing workloads running on thousands of GPUs
• Working closely with leading compiler, library, and systems teams

Requirements:

• Bachelor's, Master's, or Ph.D. in Computer Science or a related technical field
• 8+ years of relevant work experience
• Strong command of Python and experience building complex software systems
• Hands-on experience with deep learning frameworks like PyTorch or JAX
• Solid foundation in compiler concepts such as ASTs, intermediate representations, program analysis, and code generation
• Excellent communication and collaboration skills

Preferred qualifications:

• Previous contributions to deep learning compiler projects or frameworks
• Deep expertise in PyTorch's compiler stack
• Experience with JAX-like functional transformations
• Familiarity with parallel programming, distributed systems, and high-performance CUDA code
• Track record of impactful participation in open-source communities

NVIDIA offers highly competitive salaries, an extensive benefits package, and a work environment that promotes diversity, inclusion, and flexibility.