Taro Logo

Senior Software Engineer, Deep Learning Inference

NVIDIA is the world leader in accelerated computing, pioneering solutions in AI and digital twins.
$184,000 - $287,500
Machine Learning
Senior Software Engineer
Hybrid
5,000+ Employees
6+ years of experience
AI

Description For Senior Software Engineer, Deep Learning Inference

NVIDIA, the world leader in accelerated computing, is seeking a Senior Software Engineer to join their Deep Learning Inference team. This role presents an exciting opportunity to make a significant impact in the field of Deep Learning by developing state-of-the-art inference frameworks for Large Language Models (LLMs) on NVIDIA GPUs.

The position involves working with TensorRT-LLM, NVIDIA's premier library for optimizing LLM inference performance. You'll be at the forefront of AI technology, collaborating with deep learning experts, GPU architects, and DevOps engineers while engaging with the broader deep learning community through open-source development.

The ideal candidate brings 6+ years of software development experience, strong Python skills, and a deep understanding of Machine Learning concepts, particularly in Large Language Models. Experience with C++, open-source development, and frameworks like vLLM, TensorRT, PyTorch, or JAX would be highly valuable.

NVIDIA offers a competitive compensation package with a base salary range of $184,000 - $287,500 USD, plus equity and benefits. The company is known for its innovative culture and commitment to pushing technological boundaries. Working at NVIDIA means joining one of technology's most desirable employers, where you'll help build the computing platforms driving success in AI and digital twins.

This hybrid role is based in Santa Clara, CA, offering the flexibility of both remote and office work. NVIDIA maintains a strong commitment to diversity and inclusion, fostering an environment where creativity and autonomy are highly valued. If you're passionate about deep learning, high-performance computing, and want to work with some of the industry's brightest minds, this role presents an exceptional opportunity to advance your career while contributing to groundbreaking technological developments.

Last updated 34 minutes ago

Responsibilities For Senior Software Engineer, Deep Learning Inference

  • Develop components of TensorRT-LLM, NVIDIA's best-in-class library for optimizing LLM inference performance on NVIDIA GPUs
  • Provide expert solutions to internal and external TensorRT-LLM users across GitHub and internal forums
  • Help manage TensorRT-LLM's Open Source Software (OSS) repo on GitHub
  • Collaborate across diverse teams of deep learning experts, GPU architects and DevOps engineers

Requirements For Senior Software Engineer, Deep Learning Inference

Python
  • Bachelor's, Master's, PhD or equivalent experience in Computer Science, Computer Engineering, Electrical Engineering or related field
  • 6+ years of software development experience
  • Strong experience with Python
  • Strong grasp of Machine Learning concepts, especially related to Large Language Models
  • Excellent communication skills, and an aptitude for collaboration and teamwork

Benefits For Senior Software Engineer, Deep Learning Inference

Equity
  • Equity
  • Benefits package

Interested in this job?

Jobs Related To NVIDIA Senior Software Engineer, Deep Learning Inference

Senior Software Engineer, Digital Human Technology

Senior Software Engineer role at NVIDIA focusing on Digital Human Technology, involving AI, machine learning, and character animation.

Senior Software Engineer, Robotics Foundation Models

Senior Software Engineer position at NVIDIA focusing on robotics foundation models and AI, combining simulation, world models, and foundation models for humanoid robots.

Senior Deep Learning Software Engineer, Recipe Pathfinding

Senior Deep Learning Software Engineer role at NVIDIA focusing on developing software systems for LLM optimization through recipe pathfinding and efficiency improvements.

Senior Software Engineer, Digital Human Technology

Senior Software Engineer role at NVIDIA focusing on Digital Human Technology, involving AI, machine learning, and character animation.

Senior Software Engineer, Digital Human Technology

Senior Software Engineer position at NVIDIA focusing on Digital Human Technology, involving AI, machine learning, and high-performance computing.