Senior Deep Learning Compiler Engineer, XLA

Nvidia

Confirmed live yesterday Trusted
Remote

Quick summary

Work type
Remote
Location
Remote
Salary
$152,000–$241,500 / yr
Posted
28 days ago
Freshness
Confirmed live yesterday

Market check

Salary context

Below market

How this pay compares to similar roles

Similar $226k
This role $197k
$137k most similar roles pay here $293k

This role pays less than 68% of similar roles. Most pay $196,312–$254,750 — the shaded band above. At the midpoint, this role pays about $197k versus about $226k for comparable roles.

Based on 240 similar postings.

Employer

About Nvidia

Nvidia is a leading designer of graphics processing units (GPUs) and system-on-chip units, powering gaming, professional visualization, data centers, and artificial intelligence workloads. Industry: Semiconductors & AI Computing

Nvidia currently has 896 open roles on FindRole.

Listed pay typically runs $184,000–$287,500 across 876 roles with salary data.

Most-posted roles

View all roles at Nvidia

At a glance

TL;DR · Senior Deep Learning Compiler Engineer, XLA

As a Senior Deep Learning Compiler Engineer - XLA on the XLA team, you will develop compiler optimization algorithms for deep learning workloads to improve inference and training performance for the JAX framework and OpenXLA compiler on NVIDIA GPUs at scale. Your daily responsibilities include crafting optimization techniques for network graphs, designing graph partitioning and tensor sharding for distributed systems, performing performance tuning, and generating code for GPU backends. You will utilize technologies including C/C++, MLIR, LLVM, OpenAI Triton, XLA, TVM, CUDA, and OpenCL. The role focuses on the technical challenge of accelerating next-generation deep learning software by collaborating with hardware architecture teams to design compiler features. You will solve complex problems related to high-performance computing and distributed programming while implementing user-facing features in JAX and associated libraries for large-scale AI systems.

What you'll do

  • Develop compiler optimization algorithms for deep learning workloads.
  • Optimize inference and training performance for JAX and OpenXLA on NVIDIA GPUs.
  • Implement compiler optimization techniques for deep learning network graphs.
  • Design novel graph partitioning and tensor sharding techniques for distributed systems.
  • Perform performance tuning and analysis of software and hardware.
  • Generate code for GPU backends using MLIR, LLVM, and OpenAI Triton.
  • Design user-facing features in JAX and related libraries.
  • Collaborate with hardware teams to design compiler features for next-generation GPUs.

What we're looking for

  • Bachelor's, Master's, or Ph.D. in Computer Science, Computer Engineering, or a related field (or equivalent experience).
  • 4+ years of relevant work or research experience in performance analysis and compiler optimizations.
  • Excellent C/C++ programming skills including debugging, performance analysis, and test design.
  • Strong foundation in the architecture of CPUs, GPUs, or other high-performance hardware accelerators.
  • Knowledge of high-performance computing and distributed programming.
  • CUDA or OpenCL programming experience (preferred).
  • Experience with XLA, TVM, MLIR, LLVM, OpenAI Triton, deep learning models, and framework design (preferred).
  • Experience working with deep learning frameworks such as JAX, PyTorch, or TensorFlow (preferred).

More like this

Similar roles

Senior Deep Learning Compiler Engineer

Nvidia

Remote (Santa Clara, CA) +1 134 days ago $152,000$241,500
C++ Python CUDA OpenCL MLIR XLA TVM LLVM PyTorch GPU Architecture Compiler Optimization Deep Learning Kernel Generation cross compilation Performance Analysis
3+ yrs exp Remote

Deep Learning Compiler Engineer

Nvidia

Remote (Santa Clara, CA) +1 19 days ago $152,000$241,500
CUDA C++ C MLIR LLVM XLA TVM Compiler Optimization GPU Architecture Performance Analysis Deep Learning IR Design OpenCL
3+ yrs exp Remote

Senior AI Compiler Engineer, MLIR

Nvidia

Remote (Santa Clara, CA) +1 28 days ago $152,000$241,500
MLIR LLVM XLA C++ Python CUDA OpenCL PyTorch JAX GPU Architecture Kernel Generation Compiler Optimization Performance Analysis Deep Learning
3+ yrs exp Remote

Senior DL Compiler Engineer, CUDA Tile

Nvidia

Remote (Santa Clara, CA) +4 9 days ago $152,000$241,500
CUDA C++ C MLIR LLVM XLA TVM Compiler Optimization GPU Architecture Performance Analysis Deep Learning IR Design OpenCL
3+ yrs exp Remote

Senior Deep Learning Frameworks CUDA Software Engineer

Nvidia

Remote (Santa Clara, CA) +1 11 days ago $184,000$287,500
CUDA PyTorch JAX C++ Python TRT-LLM vLLM SGLang TensorRT Triton NCCL MPI UCX XLA HPC Kernel Authoring NVIDIA Nsight Systems Compiler Technologies
8+ yrs exp Remote

Senior Software Engineer, CUDA Deep Learning Systems

Nvidia

Remote (Santa Clara, CA) +1 11 days ago $184,000$287,500
CUDA C++ Python Deep Learning PyTorch JAX TensorRT vLLM Triton XLA NCCL MPI UCX Distributed Computing Kernel Optimization Model Optimization FP8 INT8
8+ yrs exp Remote