Senior Deep Learning Compiler Engineer - CUDA

at Nvidia
📍 Seattle, United States
USD 148,000-287,500 per year
SENIOR
✅ On-site

SCRAPED

Used Tools & Technologies

Not specified

Required Skills & Competences ?

Python @ 7 Machine Learning @ 4 PyTorch @ 4 CUDA @ 7 GPU @ 4

Details

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for over 25 years. The company is pioneering the next era of computing by leveraging AI, where GPUs serve as the brains of computers, robots, and self-driving cars that can understand the world. This role offers an opportunity to join a diverse and supportive environment working on groundbreaking technologies in machine learning compilers and AI systems.

Responsibilities

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

Requirements

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

Ways to Stand Out

  • Proficiency in Python and C++ for DSL development and low-level compiler internals
  • Experience with compiler frameworks such as Triton, LLVM, MLIR, TVM and IR design for GPUs
  • Deep understanding of GPU architectures (CUDA cores, Tensor Cores, memory hierarchies) and tile-based execution models
  • Strong experience in machine learning systems research and productionization
  • Open source project ownership or contributions

Benefits

  • Highly competitive salary
  • Equity and comprehensive benefits package
  • Working in cutting-edge fields including Deep Learning, AI, Autonomous Vehicles, and Virtual Reality
  • A diverse and talented professional team fostering creativity and autonomy