Senior AI Compiler Engineer, MLIR

Nvidia

Remote

Quick summary

Work type
Remote
Location
Santa Clara, CA · Austin, TX
Salary
$152,000–$241,500 / yr
Posted
46 days ago
Nearby
99+ roles within 25 mi

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $211k
This role $197k
$139k most similar roles pay here $270k

This role pays less than 60% of similar roles. Most pay $175,700–$246,150 — the shaded band above. At the midpoint, this role pays about $197k versus about $211k 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 985 open roles on FindRole.

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

Most-posted roles

View all roles at Nvidia

At a glance

TL;DR · Senior AI Compiler Engineer, MLIR

Apply Now Log in to save

NVIDIA seeks a Senior AI Compiler Engineer to join its cutting-edge team, focusing on developing an MLIR-based AI compiler for NVIDIA’s inference engine. This role involves creating graph representations and optimizations for future GPU architectures, collaborating with framework and hardware teams to enable new model patterns, defining APIs and dialects, conducting performance analysis, implementing compiler optimizations, and generating kernels for neural networks. The ideal candidate holds a degree in Computer Science or related field and possesses expertise in MLIR, XLA, LLVM, C/C++, Python, and GPU programming with CUDA or OpenCL. Additionally, knowledge of deep learning models, algorithms, and frameworks like PyTorch and JAX is essential, along with strong interpersonal skills for working in a fast-paced environment.

Skills

MLIR XLA LLVM C++ Python PyTorch JAX CUDA OpenCL

What you'll do

  • Develop MLIR-based graph representations for future GPU architectures.
  • Optimize MLIR dialects and APIs to enhance compiler performance.
  • Implement compiler optimizations and kernel generation for neural networks.
  • Collaborate with hardware teams to enable new model patterns on GPUs.
  • Conduct performance analysis and debugging of AI compilers.

What we're looking for

  • Bachelor's degree in Computer Science or equivalent experience.
  • Expertise in MLIR, XLA, LLVM compiler technologies.
  • Proficient in C/C++, Python for software design and optimization.
  • Experience with GPU kernel generation for high performance.
  • Understanding of deep learning models and frameworks like PyTorch.

Related searches

More Senior AI Compiler Engineer jobs Senior AI Compiler Engineer jobs in Santa Clara Remote Senior AI Compiler Engineer jobs All jobs at Nvidia

More like this

Similar roles

Senior AI Inference Compiler Engineer

Nvidia

Remote (Santa Clara, CA) 103 days ago $152,000–$241,500
MLIR XLA LLVM PyTorch GPU CUDA C++ Compiler Technologies Deep Learning Models LLM Inference Optimizations High Performance Computing Fast Build Time Kernel Generation Neural Networks Software Engineering
Remote
Save

Senior AI Compiler Engineer

Nvidia

Remote (Austin, TX) 26 days ago $184,000–$287,500
Python C/C++ Julia Lisp LLVM GPU reinforcement_learning genetic_algorithms predictive_modeling complex_systems AI ML optimization_passes code_generation frontend_integration
Remote
Save

Senior Compiler Engineer - AI

Nvidia

Remote (Austin, TX) 23 days ago $184,000–$287,500
Python C/C++ LLVM MLIR Reinforcement_learning Genetic_algorithms Predictive_modeling LLMs CI/CD GPU_architecture Scalability Reliability Performance_engineering Open_source_contribution
Remote
Save

Senior AI/ML Engineer

General Motors (GM)

Remote (Mountain View, CA) 5 days ago $170,600–$261,300
Python Transformers Generative_AI Multimodal_Systems AutoML Quantization Model_Distillation Architecture_Search CVPR ICML NeurIPS IJCAI KDD Robotics_Conference_Papers AV_ADAS_Experience
Remote Hybrid
Save

Senior AI/ML Engineer

Uber

Seattle, WA 10 days ago $202,000–$202,000
Python PyTorch TensorFlow XGBoost LightGBM Kafka Pinot Hive Cassandra Spark Flink CI/CD AWS Kubernetes
Save