Senior Compiler Engineer, AI Inference Platforms
Core
Building the Deep Learning & AI Compiler (DLC) to deliver leading inference performance, fast build time, and reduced memory footprints for AI workloads across data centers, personal devices, automotive, and robotics.
Role type
Senior IC compiler engineer (AI inference)
Builds
Ahead-of-Time and Just-in-Time compilers for deep learning frameworks and future NVIDIA GPUs
Domain
AI computing, deep learning, GPU architecture
Deliverable
production ML models
Required skills
compiler optimization algorithms, C/C++, Python, performance analysis, software design, debugging, test design
Preferred skills
CPU/GPU architecture, CUDA/OpenCL, deep learning frameworks (PyTorch, JAX), GPU kernel authoring, Nsight Compute, hardware bring-up, mentoring
Technologies
MLIR, LLVM, XLA, Triton, CUDA, OpenCL, PyTorch, JAX, Nsight Compute
Responsibilities
Analyze deep learning networks and develop compiler optimization algorithms; Collaborate with deep learning software framework and GPU architecture teams; Define public APIs and implement compiler techniques for AI workloads
Seniority
Senior, hands-on IC
