Senior Deep Learning Compiler Engineer - XLA

NVIDIA Gruppe; Santa Clara (CA); On-site; USD 152,000 - 241,500; Full time

Benefits offered by this job

Equity

Generous benefits package

Job summary

NVIDIA Gruppe is seeking versatile software engineers for their XLA team in Santa Clara, California. The role involves developing compiler optimization algorithms for deep learning workloads and collaborating with teams to accelerate AI systems.

Candidates must have a strong background in C/C++ and 4+ years in compiler optimizations. A competitive salary of $152,000 to $241,500 is offered, along with equity and benefits.

Applications will be accepted until March 1, 2026.

Qualifications

• 4+ years of relevant work or research experience in performance analysis and compiler optimizations.
• Ability to work independently and lead your own development efforts.
• Strong foundation in CPU and GPU architecture.

Responsibilities

• Develop compiler optimization algorithms for deep learning workloads.
• Collaborate with partners in deep learning framework teams.
• Performance tuning and analysis on NVIDIA GPUs.

Skills

C/C++ programming

Performance analysis

Compiler optimizations

CUDA or OpenCL

Deep learning frameworks

Education

Bachelor’s, Master’s or Ph.D. in Computer Science or related field

Tools

MLIR

LLVM

OpenAI Triton

Job description

NVIDIA's invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning ignited modern AI — the next era of computing — with the GPU acting as the brain of computers, robots, and self-driving cars that can perceive and understand the world. Today, we are increasingly known as “the AI computing company”.

We are looking for versatile software engineers for our XLA team. NVIDIA is at the center for the AI revolution that is transforming how people live, work, and interact with technology. Come join us to build high-performance, production‑grade software that is at the core of next‑generation AI systems.

Responsibilities

In this role, develop compiler optimization algorithms for deep learning workloads. You will optimize inference and training performance for the JAX framework and the OpenXLA compiler on NVIDIA GPUs at scale. You’ll collaborate with our partners in deep learning framework teams and our hardware architecture teams to accelerate the next generation of deep learning software. The scope of these efforts includes:

• Crafting and implementing compiler optimization techniques for deep learning network graphs.
• Designing novel graph partitioning and tensor sharding techniques for distributed training and inference.
• Performance tuning and analysis.
• Code‑generation for NVIDIA GPU backends using open‑source compilers such as MLIR, LLVM and OpenAI Triton.
• Designing user‑facing features in JAX and related libraries and other general software engineering work.
• Working closely with GPU hardware engineering teams to design AI compiler software features for next‑generation GPUs.

Qualifications

• Bachelor’s, Master’s or Ph.D. in Computer Science, Computer Engineering or related field (or equivalent experience).
• 4+ years of relevant work or research experience in performance analysis and compiler optimizations.
• Ability to work independently, define project goals and scope, and lead your own development effort adopting clean software engineering and testing practices.
• Excellent C/C++ programming and software design skills, including debugging, performance analysis, and test design.
• Strong foundation in architecture of CPU, GPUs or other high‑performance hardware accelerators. Knowledge of high‑performance computing and distributed programming.
• CUDA or OpenCL programming experience is desired but not required.
• Experience with the following technologies is a huge plus: XLA, TVM, MLIR, LLVM, OpenAI Triton, deep learning models and algorithms, and deep learning framework design.
• Strong interpersonal skills are required along with the ability to work in a dynamic product‑oriented team. A history of mentoring junior engineers and interns is a bonus.

Ways to stand out from the crowd

• Experience working deep learning frameworks such as JAX, PyTorch or TensorFlow.
• Extensive experience with CUDA or with GPUs in general.
• Experience with open‑source compilers such as XLA, LLVM, MLIR or TVM.

With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world’s most desirable employers. We have some of the most forward‑thinking and hardworking people in the world working for us, and due to unprecedented growth, our exclusive engineering teams are rapidly expanding. If you're a creative and autonomous engineer with a real passion for technology, we want to hear from you.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 152,000 USD – 241,500 USD.

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until March 1, 2026.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.