Deep Learning Compiler Engineer

Nvidia

Confirmed live today Low trust
Remote

Quick summary

Work type
Remote
Location
Santa Clara, CAAustin, TXORWACARedmond, WA
Salary
$152,000–$241,500 / yr
Employment
Full-time
Posted
65 days ago
Freshness
Confirmed live today

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $215k
This role $197k
$139k $277k
below market most similar roles pay here above market

This role pays less than 63% of similar roles. Most pay $188,675–$241,750 — the blue band above. At the midpoint, this role pays about $197k versus about $215k 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 1623 open roles on FindRole.

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

Most-posted roles

View all roles at Nvidia

At a glance

TL;DR · Deep Learning Compiler Engineer

As a Deep Learning Compiler Engineer, you will join the Tensor IR & CUDA Tile team to develop compiler technologies for NVIDIA GPUs. You will design and implement compiler transformations, develop MLIR-based dialects and lowering passes, and optimize the performance of tile-based kernels to ensure efficient execution across multiple GPU architectures. Your daily work involves defining public APIs, crafting optimization techniques, and performing general software engineering tasks. You will utilize C/C++ for programming, debugging, and performance analysis. The role requires expertise in compiler optimization, IR design, and performance analysis. You may also leverage knowledge of LLVM, XLA, TVM, CUDA, or OpenCL. This position addresses the technical challenge of generating high-performance GPU kernels for deep learning models and generative AI applications.

What you'll do

  • Design and implement compiler transformations for CUDA Tile and TensorIR technologies.
  • Develop MLIR-based dialects and lowering passes for NVIDIA GPUs.
  • Optimize the performance of tile-based kernels across multiple GPU architectures.
  • Define public APIs for compiler and optimization techniques.
  • Perform performance analysis and debugging for high-performance GPU kernels.
  • Execute general software engineering tasks to support deep learning infrastructure.

What we're looking for

  • Bachelor's, Master's, or Ph.D. in Computer Science, Computer Engineering, or a related field (or equivalent experience).
  • 3+ years of relevant work or research experience in compiler optimization, performance analysis, and IR design.
  • Excellent C/C++ programming and software design skills, including debugging, performance analysis, and test design.
  • Ability to work independently, define project goals and scope, and lead your own development effort.
  • Strong interpersonal skills and the ability to work in a dynamic product-oriented team.
  • Knowledge of CPU and/or GPU architecture (preferred).
  • CUDA or OpenCL programming experience (preferred).
  • Experience with MLIR, LLVM, XLA, TVM, and deep learning models and algorithms (preferred).

More like this

Similar roles

Senior Deep Learning Compiler Engineer

Nvidia

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

Senior Deep Learning Compiler Engineer, HW-SW Codesign

Nvidia

Remote (Santa Clara, CA) +2 42 days ago $152,000–$241,500
C++ CUDA XLA MLIR LLVM OpenAI Triton JAX PyTorch TensorFlow TVM OpenCL Distributed Programming High-Performance Computing Compiler Optimization Deep Learning Frameworks
4+ yrs exp Remote

Senior Deep Learning Compiler Engineer, XLA

Nvidia

Remote (Santa Clara, CA) +5 41 days ago
C++ XLA MLIR LLVM CUDA OpenAI Triton JAX PyTorch TensorFlow TVM OpenCL Distributed Programming High-Performance Computing Compiler Optimization Deep Learning Frameworks
4+ yrs exp Remote

Senior AI Compiler Engineer, MLIR

Nvidia

Remote (Santa Clara, CA) +2 74 days ago
MLIR LLVM XLA C++ Python CUDA OpenCL PyTorch JAX GPU Architecture Compiler Optimizations Kernel Generation Deep Learning Performance Analysis
3+ yrs exp Remote

Senior DL Compiler Engineer, CUDA Tile

Nvidia

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