AI Compiler Engineer

NVIDIA·Santa Clara, CA

AI EngineeringSeniorFull-time

$190K-$330KPosted 1 months ago

About the Role

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.

Requirements

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

Nice to Have

Experience with Triton or OpenAI Triton
Background in polyhedral compilation
Experience with MLIR
Familiarity with AI-specific hardware architectures

Benefits

NVIDIA stock awards

Premium benefits package

Patent bonus program

On-site fitness center

Education reimbursement

Employee stock purchase plan

Skills

CompilersCUDAC++LLVMPythonGPU Optimization

Apply for this Position

Related Jobs

Prompt Engineer

xAI · Palo Alto, CA

$150K-$250KAI Engineering

Open Source ML Engineer

Hugging Face · New York, NY · Remote

$180K-$280KAI Engineering

CUDA Engineer — AI Frameworks

NVIDIA · Santa Clara, CA

$200K-$350KAI Engineering

AI Developer Relations Engineer

NVIDIA · Austin, TX · Remote

$150K-$250KSales & Marketing

Preparing for Your AI Career?

Vincony has all 400+ AI models in one place — compare responses, AI debate, Image/Video/Voice generator, and 20 more tools to help you learn and build with AI.

Visit Vincony.com