Senior ML Inference Engineer - GPUs & TensorRT

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

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA in California is seeking an experienced Senior Software Engineer to join the TensorRT team, shaping the next generation of inference software for AI accelerators. You will design and optimize TensorRT and TensorRT-LLM, write in C++, Python and CUDA, and work with GPU architects to push performance on modern GPUs.

This role requires advanced degrees and 4+ years of experience, offers equity and benefits, and a clear path to impactful work on real-time AI inference.

Qualifications

• BS, MS, PhD or equivalent in Computer Science, Computer Engineering or 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.

Responsibilities

• 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.

Skills

C++

Python

Rust

CUDA

Deep Learning Frameworks

Education

BS/MS/PhD in Computer Science or Computer Engineering or related field

Tools

TensorRT

Job description

NVIDIA in California is seeking an experienced Senior Software Engineer to join the TensorRT team, shaping the next generation of inference software for AI accelerators. You will design and optimize TensorRT and TensorRT-LLM, write in C++, Python and CUDA, and work with GPU architects to push performance on modern GPUs.

This role requires advanced degrees and 4+ years of experience, offers equity and benefits, and a clear path to impactful work on real-time AI inference.