Senior Deep Learning Compiler Engineer

Nvidia

Remote Actively hiring
Remote · Santa Clara, CA · Austin, TX · Remote, WA Posted 29 days ago $152,000–$241,500 / year
View original post Log in to save

At a glance

AI generated

TL;DR

NVIDIA seeks a Deep Learning Compiler Engineer to join its DLC team, focusing on advancing the inference engine for applications like large language models and image classification. This role involves collaborating with software framework and hardware architecture teams to enhance public APIs, optimize performance, and develop compiler infrastructure for neural networks. The ideal candidate should have expertise in CPU and GPU architectures, experience with constrained resource systems, and proficiency in technologies such as MLIR, XLA, TVM, LLVM, and deep learning frameworks like PyTorch. Additionally, the ability to generate high-performance GPU kernels and mentor junior engineers is highly valued. This position plays a crucial role in delivering leading inference performance across data centers, personal devices, automotive, and robotics sectors.

Skills

MLIR XLA TVM LLVM PyTorch CUDA C++ Python GPU CPU Embedded_Systems Cross_Compilation CI/CD

What you'll do

  • Define public APIs for the Deep Learning Compiler to enhance usability.
  • Optimize compiler infrastructure techniques for neural networks to improve performance.
  • Implement performance optimizations and analysis for deep learning software frameworks.
  • Develop GPU kernel generation with high performance and fast build time.
  • Work on reducing memory footprints in systems with constrained resources.

What we're looking for

  • Bachelor’s degree or higher in computer science or related field.
  • Proficiency in CPU and GPU architecture optimization.
  • Experience working on systems with constrained resources like embedded platforms.
  • Expertise in MLIR, XLA, TVM, LLVM, and deep learning frameworks such as PyTorch.
  • Ability to generate high-performance GPU kernels efficiently.
  • Track record of mentoring junior engineers and interns (preferred).

Market check

Salary context

This $152,000–$241,500 range sits above 47% of similar postings on FindRole.

Peer median band

$155,420–$241,500

Median floor and ceiling across peers.

Typical midpoint (25–75%)

$179,493–$235,750

Middle half of comparable postings.

Based on 240 comparable postings.

* 240 is the maximum number of comparable postings sampled.

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 801 open roles on FindRole.

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

Most-posted roles

View all roles at Nvidia

More like this

Similar roles

Senior Deep Learning Compiler Engineer - XLA

Nvidia

Remote (Us, Ca, Santa Clara, US) 93 days ago $152,000–$241,500
C/C++ CUDA JAX PyTorch TensorFlow XLA MLIR LLVM OpenAI_Triton GPU distributed_programming performance_analysis compiler_optimizations clean_software_engineering_practices high_performance_computing
Remote

Senior Deep Learning Software Engineer

Nvidia

US 85 days ago $224,000–$356,500
Python PyTorch JAX CUDA TensorRT NVIDIA_TensorRT_LLM GPU_optimization CUTLASS Triton Deep_learning_frameworks Performance_analysis GPU_architecture High_performance_computing Model_inference Inference_optimization