Senior Deep Learning Compiler Engineer, HW-SW Codesign
Nvidia
Quick summary
Market check
How this pay compares to similar roles
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
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
At a glance
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.
Skills
What you'll do
What we're looking for
More like this
Nvidia
Nvidia
Nvidia
Nvidia
Nvidia
Nvidia