Senior Deep Learning Compiler Engineer, XLA

Nvidia

Confirmed live yesterday High trust
Remote

Quick summary

Work type
Remote
Location
Santa Clara, CAAustin, TX
Salary
$152,000–$241,500 / yr
Posted
3 days ago
Freshness
Confirmed live yesterday

Market check

Salary context

Below market

How this pay compares to similar roles

Similar $224k
This role $197k
$138k most similar roles pay here $285k

This role pays less than 68% of similar roles. Most pay $194,000–$254,750 — the shaded band above. At the midpoint, this role pays about $197k versus about $224k 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 910 open roles on FindRole.

Listed pay typically runs $184,000–$287,500 across 894 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, you will join the XLA team to develop compiler optimization algorithms for deep learning workloads. You will focus on optimizing inference and training performance for the JAX framework and the 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, and TVM while collaborating with hardware architecture teams to design features for next-generation GPUs. The role addresses the technical challenge of accelerating deep learning software by improving compiler performance and designing user-facing features in JAX and related libraries within a high-performance computing environment.

What you'll do

  • Develop compiler optimization algorithms for deep learning workloads on NVIDIA GPUs.
  • Optimize inference and training performance for the JAX framework and OpenXLA compiler.
  • Implement compiler optimization techniques for deep learning network graphs.
  • Design novel graph partitioning and tensor sharding techniques for distributed training and inference.
  • Perform performance tuning and analysis for high-performance computing systems.
  • Generate code for NVIDIA GPU backends using MLIR, LLVM, and OpenAI Triton.
  • Design user-facing features in JAX and related libraries.
  • Collaborate with hardware engineering teams to design software 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, algorithms, 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 142 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 27 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 36 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) +1 10 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 DL Compiler Engineer, CUDA Tile

Nvidia

Remote (Santa Clara, CA) +4 17 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