Senior AI Compiler Engineer — GPU Performance Optimizations

Best AI Tools Wiki; Santa Clara (CA); Hybrid; USD 180,000 - 320,000; Full time

Benefits offered by this job

NVIDIA stock awards

Premium benefits package

On-site fitness center

Employee stock purchase plan

Job summary

NVIDIA is seeking an AI Compiler Engineer to develop optimizations for deep learning workloads. You will work on Triton, XLA, and CUDA compilers to generate high-performance code for NVIDIA GPUs, enabling faster training and inference for AI models.

This on-site role in California offers stock awards, premium benefits, and an employee stock purchase plan. You will leverage strong C++/Python, LLVM, and GPU architecture knowledge to drive code generation and optimization across cutting-edge AI

Qualifications

• 5+ years of compiler engineering experience.
• Strong C++ and Python skills.
• Experience with LLVM or similar compiler frameworks.
• Understanding of GPU architecture and memory hierarchy.
• Background in code generation and optimization.

Responsibilities

• Develop optimizations for deep learning workloads.
• Work on Triton, XLA, and CUDA compilers to generate high-performance code.
• Enable faster training and inference for AI models.

Skills

Compiler engineering

C++

Python

LLVM

GPU architecture

Tools

LLVM

CUDA

MLIR

Triton

Job description

NVIDIA is seeking an AI Compiler Engineer to develop optimizations for deep learning workloads. You will work on Triton, XLA, and CUDA compilers to generate high-performance code for NVIDIA GPUs, enabling faster training and inference for AI models.

This on-site role in California offers stock awards, premium benefits, and an employee stock purchase plan. You will leverage strong C++/Python, LLVM, and GPU architecture knowledge to drive code generation and optimization across cutting-edge AI