Senior AI Compiler Engineer, Algorithms and Code-Generation

NVIDIA
Santa Clara, Austin
Workplace: HybridFull timeUSD 152,000 - 241,500 annuallyFunction: Data Science & Machine LearningEducation: bachelorsSkills: ["Independent work","Interpersonal skills","Team collaboration"]

Build and optimize NVIDIA’s Deep Learning & AI Compiler (DLC) for leading inference performance across data centers, devices, automotive, and robotics. Analyze deep learning networks to develop compiler optimization algorithms, define public APIs, and craft compiler techniques for AI workloads and future GPUs. Work with CUDA to debug performance bottlenecks, advance Ahead-of-Time and Just-in-Time compilation, and improve memory footprints and build times.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
NVIDIA
NVIDIA
2 months ago

Senior AI Compiler Engineer, Algorithms and Code-Generation

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 16 hours agoStatus: Live
Reposted: similar role first listed 4 months ago

Job Summary

Build and optimize NVIDIA’s Deep Learning & AI Compiler (DLC) for leading inference performance across data centers, devices, automotive, and robotics. Analyze deep learning networks to develop compiler optimization algorithms, define public APIs, and craft compiler techniques for AI workloads and future GPUs. Work with CUDA to debug performance bottlenecks, advance Ahead-of-Time and Just-in-Time compilation, and improve memory footprints and build times.
Location: Santa Clara, Austin
Workplace: Hybrid
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Mid level

Key Responsibilities

  • •Analyze deep learning networks and develop compiler optimization algorithms.
  • •Use strong CUDA programming skills to analyze and debug GPU performance bottlenecks.
  • •Define public APIs and implement performance optimizations and analysis for AI workloads.
  • •Craft and implement compiler techniques for AI workloads and future NVIDIA GPUs in both Ahead-of-Time and Just-in-Time forms.

Pay and Benefits

Salary: USD 152,000 - 241,500 annually

Key Requirements

  • •Bachelor’s, master’s, or Ph.D. in Computer Science, Computer Engineering, related field, or equivalent experience.
  • •3+ years of relevant work or research experience in performance analysis and compiler optimizations.
  • •Experience with compiler technologies such as MLIR, LLVM, XLA, and Triton (or equivalents).
  • •Strong C/C++ and Python 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 efforts.
Experience:Deep learningAIGPU computingCompiler optimization
Education:Bachelor's
Skills:Independent workInterpersonal skillsTeam collaboration
Tech Stack:CUDAC++CPythonMLIRLLVMXLATritonPyTorchJAXNsight Compute

Company Brief

NVIDIA
Designs and manufactures GPUs, AI accelerators, and system-on-chip products for gaming, data centers, professional visualization, and automotive markets, enabling advanced graphics, AI, and high-performance computing solutions worldwide.
Industry: Electronics Manufacturing
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: Santa Clara, United States
Founded: 1993
Glassdoor
Glassdoor: 4.3
WebsiteLinkedInGlassdoor