Senior Software Engineer, Deep Learning - MLIR TRT

Reposted 2 Days Ago
Be an Early Applicant
Santa Clara, CA
In-Office
184K-357K Annually
Senior level
Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
The Role
Develop and optimize deep learning compiler technologies for NVIDIA's hardware in autonomous vehicles. Collaborate with partners to improve model performance and deployment.
Summary Generated by Built In

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It’s a unique legacy of innovation that’s fueled by great technology—and amazing people. Today, we’re tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what’s never been done before takes vision, innovation, and the world’s best talent. As an NVIDIAN, you’ll be immersed in a diverse, supportive environment where everyone is inspired to do their best work. Come join the team and see how you can make a lasting impact on the world.

We are looking for outstanding Senior Deep Learning Software Engineers to develop and productize NVIDIA's deep learning solutions in autonomous driving vehicles. In the Solution Engineering-Automotive Machine Learning team, we are developing new technologies to allow more capable deep learning models to be deployed in Physical AI systems. As part of the role, you will develop compiler technology to allow larger and better models to be optimized to leverage NVIDIA’s unique hardware architecture. You will also be exposed to the most pressing problems that our partners face during product development and coordinate with other architecture and software teams to develop the best solution for partners working on our platforms.

What you'll be doing:

  • Developing compiler technologies to accelerate deep learning inference on NVIDIA hardware platforms for Physical AI.

  • Working across a wide range of abstractions from model fine-tuning and quantization to low-level kernel development and performance optimization.

  • Develop workflows that let users leverage frameworks (e.g. PyTorch, JAX) and compiler technologies tools (e.g. MLIR, Triton) without forgoing performance

  • Work with customers to help accelerate their workloads on NVIDIA platforms.

  • Stay up to date with the latest research and innovations in deep learning, implement and experiment with new insights to improve NVIDIA's Physical AI DNNs.

What we need to see:

  • MS or PhD degree in computer science, computer vision, robotics, computer architecture or equivalent experience in technical field (or equivalent experience)

  • 5+ years of work experience in software development.

  • 2+ years of experience in **developing** deep learning frameworks (e.g. PyTorch, JAX, TensorFlow, ONNX, etc.) or compiler technologies (e.g. LLVM, MLIR, TVM, Triton, etc.).

  • Domain experience in technologies used for GPU programming (e.g. CUDA C++ and/or DSLs like OpenAI Triton) or with system-level optimization for deep learning training or inference.

  • Strong C/C++ programming skills

  • Familiar with start-of-the-art deep learning techniques for inference and training.

  • Willing to take action and have strong analytical skills.

Ways to stand out from the crowd:

  • Experience with MLIR or LLVM or similar compiler technologies

  • Background with low precision inference, quantization, compression of DNNs

  • Experience with GPU programming

  • Experience with building DSLs or optimizing compilers (e.g. graph compiler or kernel generator) for GPUs or other accelerated computing platforms.

  • Open source project ownership or contribution, healthy GitHub repositories, guiding and/or mentoring experience

Widely considered to be one of the technology world’s most desirable employers, NVIDIA offers highly competitive salaries and a comprehensive benefits package. As you plan your future, see what we can offer to you and your family www.nvidiabenefits.com/

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5.

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until January 13, 2026.

This posting is for an existing vacancy. 

NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.

Top Skills

Cuda C++
Jax
Llvm
Mlir
Onnx
PyTorch
TensorFlow
Triton
Tvm
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Santa Clara, CA
21,960 Employees
Year Founded: 1993

What We Do

NVIDIA’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning ignited modern AI — the next era of computing — with the GPU acting as the brain of computers, robots, and self-driving cars that can perceive and understand the world. Today, NVIDIA is increasingly known as “the AI computing company.”

Similar Jobs

Metropolis Technologies Logo Metropolis Technologies

Staff Software Engineer

Artificial Intelligence • Computer Vision • Machine Learning • Payments • Real Estate • PropTech
Easy Apply
In-Office
Los Angeles, CA, USA
23100 Employees
220K-250K Annually

DigitalOcean Logo DigitalOcean

Software Engineer

Artificial Intelligence • Cloud • Software • Infrastructure as a Service (IaaS)
In-Office
San Francisco, CA, USA
1400 Employees
200K-250K Annually

Samsara Logo Samsara

Staff Software Engineer

Artificial Intelligence • Cloud • Computer Vision • Hardware • Internet of Things • Software
Easy Apply
Remote or Hybrid
6 Locations
4000 Employees
162K-290K Annually

People Inc. Logo People Inc.

Associate Director, Creative Production & Development, Travel + Leisure

AdTech • Consumer Web • Digital Media • eCommerce • Marketing Tech
Hybrid
2 Locations
3500 Employees
105K-125K Annually

Similar Companies Hiring

Milestone Systems Thumbnail
Software • Security • Other • Big Data Analytics • Artificial Intelligence • Analytics
Lake Oswego, OR
1500 Employees
Idler Thumbnail
Artificial Intelligence
San Francisco, California
6 Employees
Fairly Even Thumbnail
Software • Sales • Robotics • Other • Hospitality • Hardware
New York, NY

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account