Senior Software Engineer, Machine Learning Inference

NVIDIA; Santa Clara (CA); On-site; USD 152,000 - 288,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is seeking talented engineers for the TensorRT team to advance the industry-leading deep learning inference software. You will design, develop and optimize TensorRT/TensorRT-LLM for diverse inference workloads across data centers, workstations and PCs.

The role requires strong C++, Python, and CUDA skills, plus experience with DL frameworks and compilers. You will collaborate with DL experts and GPU architects to influence HW/SW design in a fast-paced environment.

Qualifications

• BS/MS/PhD or equivalent experience in Computer Science or related field.
• 4+ years of software development on a large codebase or project.
• Strong proficiency in C++ (required), Rust or Python.
• Experience in developing Deep Learning frameworks, compilers, or system software.
• Excellent problem-solving and collaborative skills.

Responsibilities

• Design, develop and optimize TensorRT and TensorRT-LLM for inference applications.
• Develop software in C++, Python, and CUDA for deploying state-of-the-art LLMs and Generative AI models.
• Collaborate with DL experts and GPU architects to influence hardware and software design for inference.

Skills

C++

Rust

Python

CUDA

Education

BS in CS/CE

MS in CS/CE

PhD in CS/CE

Tools

TensorRT

PyTorch

JAX

Job description

At NVIDIA, we're at the forefront of innovation, driving advancements in AI and machine learning to solve some of the world’s most challenging problems. We're seeking talented and motivated engineers to join our TensorRT team in developing the industry-leading deep learning inference software for NVIDIA AI accelerators.

What You’ll Be Doing

• Design, develop and optimize NVIDIA TensorRT and TensorRT-LLM to supercharge inference applications for datacenter, workstations, and PCs.
• Develop software in C++, Python, and CUDA for seamless and efficient deployment of state-of-the-art LLMs and Generative AI models.
• Collaborate with deep learning experts and GPU architects throughout the company to influence Hardware and Software design for inference.

What We Need To See

• BS, MS, PhD or equivalent experience in Computer Science, Computer Engineering or a related field.
• 4+ years of software development experience on a large codebase or project.
• Strong proficiency in C++ (required), Rust or Python programming languages.
• Experience in developing Deep Learning Frameworks, Compilers, or System Software.
• Excellent problem-solving skills and passion to learn and work effectively in a fast-paced, collaborative environment.
• Strong communication skills and the ability to articulate complex technical concepts.

Ways To Stand Out From The Crowd

• Experience in developing inference backends and compilers for GPUs.
• Knowledge of Machine Learning techniques and GPU programming with CUDA or OpenCL.
• Background in working with LLM inference frameworks like TensorRT-LLM, vLLM, SGLang.
• Experience working with deep learning frameworks like TensorRT, PyTorch, JAX.
• Knowledge of close-to-metal performance analysis, optimization techniques, and tools.

NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us. If you're creative, autonomous and love a challenge, we want to hear from you. Come, join our team and help build the real-time, cost-effective computing platform driving our success in this exciting and quickly growing field.

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

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until August 9, 2026.

This posting is for an existing vacancy.

NVIDIA is committed to fostering an inclusive 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.

, ,