Senior Software Engineer, Deep Learning - MLIR TRT

at Nvidia

USD 184,000-356,500 per year

SENIOR

✅ On-site

Tech Stack
Tag name is followed by "@" symbol and proficiency level value. About proficiency levels:

1-2 — basic awareness. Minimal hands-on experience, and a rudimentary understanding of the technology's purpose;

3-6 — daily use. Comfortable and regular usage, capable of handling common tasks and challenges related to the technology;

7-9 — you are an expert, you can teach others, you know all the pitfalls and tricks;

10 — exceptional knowledge, comprehensive understanding, and adeptness in all aspects of the technology, including advanced problem-solving. Think twice before claiming or demanding such level.

AI CUDA @ 4 Computer Vision @ 4 Deep Learning @ 4 GPU @ 4 GitHub @ 4 JAX @ 4 LLVM @ 4 Machine Learning Mentoring @ 4 Performance Optimization PyTorch @ 4 Robotics @ 4 Software Development @ 6 TensorFlow @ 4

Details

NVIDIA is developing deep learning solutions for autonomous driving vehicles and Physical AI systems. The Solution Engineering-Automotive Machine Learning team develops compiler technology that enables larger and more capable deep learning models to leverage NVIDIA's hardware architecture. This role involves working with architecture and software teams and addressing partner product-development challenges.

Responsibilities

Develop compiler technologies to accelerate deep learning inference on NVIDIA hardware platforms for Physical AI.
Work across abstractions ranging from model fine-tuning and quantization to low-level kernel development and performance optimization.
Develop workflows that allow users to leverage frameworks such as PyTorch and JAX, together with compiler technology tools such as MLIR and Triton, without sacrificing performance.
Work with customers to accelerate their workloads on NVIDIA platforms.
Stay current with deep learning research and innovations, and implement and experiment with new insights to improve NVIDIA's Physical AI deep neural networks.

Requirements

Master's or PhD degree in computer science, computer vision, robotics, computer architecture, or an equivalent technical field, or equivalent experience.
5+ years of software development experience.
2+ years of experience developing deep learning frameworks such as PyTorch, JAX, TensorFlow, or ONNX, or compiler technologies such as LLVM, MLIR, TVM, or Triton.
Experience with GPU programming technologies such as CUDA C++ or domain-specific languages such as OpenAI Triton, or with system-level optimization for deep learning training or inference.
Strong C/C++ programming skills.
Familiarity with state-of-the-art deep learning techniques for inference and training.
Strong analytical skills and a willingness to take action.

Preferred Qualifications

Experience with MLIR, LLVM, or similar compiler technologies.
Background in low-precision inference, quantization, or deep neural network compression.
Experience with GPU programming.
Experience building domain-specific languages or optimizing compilers, such as graph compilers or kernel generators, for GPUs or other accelerated computing platforms.
Open-source project ownership or contribution, healthy GitHub repositories, and experience guiding or mentoring others.

Benefits

NVIDIA offers competitive salaries, equity, and a comprehensive benefits package. The base salary depends on location, experience, and the pay of employees in similar positions. Applications will be accepted at least until January 13, 2026. This posting is for an existing vacancy.

NVIDIA is an equal opportunity employer committed to fostering a diverse work environment.