Senior ML Inference & Compiler Engineer — Equity Options

NVIDIA; United States; On-site; USD 152,000 - 288,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is seeking a Senior Machine Learning Applications and Compiler Engineer to develop algorithms and optimizations for our LPX inference and compiler stack. You will craft how neural network workloads map onto NVIDIA platforms, working at the intersection of large-scale systems, compilers, and deep learning.

You will extend and integrate with NVIDIA’s software ecosystem, benchmark performance, and collaborate with hardware teams to drive new features and efficiency.

Qualifications

• 5 years of relevant experience in ML, compilers, or runtime development.
• Proficiency in systems level programming (C/C++ or Rust).
• Experience with compiler or runtime development, including IR design or code generation.
• Familiarity with ML frameworks (TensorFlow, PyTorch) and graph formats.

Responsibilities

• Build, test, and maintain high-performance runtime and compiler components for end-to-end inference.
• Map large-scale inference workloads onto NVIDIA systems.
• Extend NVIDIA SW ecosystem with libraries, tooling, and interfaces for model deployment.

Skills

Systems programming

C/C++

Rust

Data structures

Algorithms

Concurrency

Education

MS/PhD in CS/EE

Tools

LLVM

MLIR

ONNX

Job description

NVIDIA is seeking a Senior Machine Learning Applications and Compiler Engineer to develop algorithms and optimizations for our LPX inference and compiler stack. You will craft how neural network workloads map onto NVIDIA platforms, working at the intersection of large-scale systems, compilers, and deep learning.

You will extend and integrate with NVIDIA’s software ecosystem, benchmark performance, and collaborate with hardware teams to drive new features and efficiency.