Senior Compiler Engineer, AI Inference Platforms
About The Position
NVIDIA is hiring a compiler engineer for its Deep Learning & AI Compiler (DLC) team. The team builds compiler technology that powers NVIDIA’s inference engine across data centers, personal devices, automotive, and robotics. The compiler supports ahead-of-time and just-in-time compilation and aims to deliver leading inference performance, fast build times, reduced memory footprints, and ease of use. The role focuses on analyzing deep learning networks and developing compiler optimization algorithms, while collaborating with deep learning software framework and GPU architecture teams. Work includes defining public APIs, performance analysis and optimization, and implementing compiler techniques for AI workloads and future NVIDIA GPUs.
Requirements
- Bachelor’s, Master’s, or Ph.D. in Computer Science, Computer Engineering, a related field, or equivalent experience.
- At least 3 years of relevant work or research experience in performance analysis and compiler optimizations.
- Experience with compiler technologies such as MLIR, LLVM, XLA, or Triton.
- Excellent C/C++ and Python programming and software design skills, including debugging, performance analysis, and test design.
- Ability to work independently, define project goals and scope, and lead development efforts.
- Strong interpersonal skills and ability to work in a dynamic, product-oriented team.
Nice To Haves
- Proficiency in CPU and/or GPU architecture.
- CUDA or OpenCL programming experience.
- Understanding of deep learning models, algorithms, and frameworks such as PyTorch and JAX.
- GPU kernel authoring and performance analysis using tools such as Nsight Compute.
- A track record of mentoring early-career engineers and interns.
- A track record of new hardware bring-up.
Responsibilities
- Analyze deep learning networks and develop compiler optimization algorithms.
- Collaborate with deep learning software framework teams and GPU architecture teams to accelerate next-generation deep learning software.
- Define public APIs and perform performance optimization and analysis.
- Craft and implement compiler techniques for AI workloads and future NVIDIA GPUs.
Benefits
- Equity
- Benefits package
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Senior
