Senior AI Compiler Engineer, Algorithms and Code Generation
NVIDIAJob Title
Senior AI Compiler Engineer, Algorithms and Code Generation
Role Summary
The engineer will design and implement compiler algorithms and code-generation techniques for deep learning workloads, focusing on performance, memory, and build-time optimizations for NVIDIA GPUs.
The role sits on the Deep Learning & AI Compiler (DLC) team and collaborates with product and research teams to deliver Ahead-of-Time and Just-in-Time compiler solutions used across data center and edge platforms.
Experience Level
Senior — typically 3+ years of relevant industry or research experience in compiler optimizations and performance analysis.
Responsibilities
Key responsibilities include:
- Analyze deep learning networks and design compiler optimization algorithms for inference workloads.
- Implement code-generation and performance optimizations targeting modern GPUs.
- Develop, define, and maintain public APIs and compiler interfaces.
- Profile and debug GPU performance bottlenecks, validate optimizations, and improve build times and memory usage.
- Collaborate with cross-functional teams to integrate compiler techniques into product pipelines.
Requirements
Must-have technical skills and experience:
- 3+ years relevant work or research experience in performance analysis and compiler optimizations.
- Experience with compiler technologies such as MLIR, LLVM, XLA, or Triton.
- Strong C/C++ and Python programming skills, including debugging, testing, and software design.
- Experience with CUDA programming and GPU performance analysis.
- Ability to work independently, define scope and deliverables, and drive projects to completion.
- Strong communication and collaboration skills for working in product-focused teams.
Nice-to-have:
- Deep knowledge of CPU/GPU architectures, especially modern NVIDIA GPU microarchitecture.
- Familiarity with deep learning frameworks (PyTorch, JAX) and model internals.
- GPU kernel authoring experience and use of tools like Nsight Compute.
- Experience mentoring junior engineers and involvement in hardware bring-up.
Education Requirements
Bachelor's, Master's, or Ph.D. in Computer Science, Computer Engineering, or a related technical field — or equivalent practical experience.
About the Company
Company: NVIDIA
Headquarters: Santa Clara, California, USA
NVIDIA is a global leader in accelerated computing, renowned for its innovative solutions in AI and digital twins that transform diverse industries. The company specializes in networking technologies, providing end-to-end InfiniBand and Ethernet solutions for servers and storage that optimize performance and scalability. NVIDIA serves sectors such as high-performance computing, enterprise data centers, and cloud computing, constantly reinventing its products and services to stay ahead in the market.
