Senior Software Engineer – TensorRT Edge-LLM

NVIDIA Corporation

Santa Clara (CA)

Hybrid

USD 152,000 - 288,000

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Benefits offered by this job

Equity
Benefits
Hybrid work model

Job summary

NVIDIA Corporation is seeking a Senior Software Engineer for the TensorRT Edge-LLM team in the US. You will develop a high-performance inference framework in modern C++ that extends TensorRT for autoregressive model serving, including speculative decoding and KV cache management, within embedded and edge platforms.

You will collaborate across CUDA and robotics teams, optimize transformer components, and contribute to kernel development while staying ahead of LLM/VLM trends.

Qualifications

  • BS, MS, PhD or equivalent experience in Computer Science, Electrical/Computer Engineering, or a closely related field.
  • 4+ years of relevant software development experience.
  • Deep understanding of transformer models and inference optimization techniques (e.g., quantization, tensor parallelism, or memory-efficient scheduling).
  • Proficient programming ability with modern C++ (C++11/14/17 and beyond).
  • Familiarity with popular LLM frameworks and libraries such as TensorRT, TensorRT-LLM, vLLM, SGLang, MLC-LLM, or FlashInfer.
  • A track record of strong software design, execution, and collaboration across fields.

Responsibilities

  • Develop and evolve a state-of-the-art inference framework in modern C++ that extends TensorRT with autoregressive model serving capabilities.
  • Design and implement compiler and runtime optimizations for transformer-based models on constrained, real-time platforms.
  • Collaborate with CUDA, kernel libraries, compilers, and robotics teams to deliver high-performance, production-ready solutions.
  • Contribute to CUDA kernel and operator development for attention, GEMM, and MoE components.
  • Benchmark, profile, and optimize inference performance across diverse embedded and automotive environments.
  • Stay ahead of evolving LLM/VLM ecosystems and bring emerging techniques into product-grade software.

Skills

Transformer models
C++
Inference optimization
Quantization
TensorRT

Education

BS/MS/PhD in CS/EE

Tools

TensorRT
TensorRT-LLM
vLLM
SGLang
MLC-LLM
FlashInfer
CUDA

Job description

## Senior Software Engineer – TensorRT Edge-LLMApplylocations: US, CA, Santa Clara: US, TX, Austin: US, CA, Remotetime type: Full timeposted on: Posted Yesterdayjob requisition id: JR2012868Are you passionate about pushing the limits of real-time large language model inference? Join NVIDIA’s TensorRT Edge-LLM team and help shape the next generation of edge AI for automotive and robotics. We build the software stack that enables Large Language, Vision-Language, and Multimodal (LLM/VLM/VLA) models to run efficiently on embedded and edge platforms — delivering cutting-edge generative AI experiences directly on-device.**What you’ll be doing:*** Develop and evolve a state-of-the-art inference framework in modern C++ that extends TensorRT with autoregressive model serving capabilities, including speculative decoding, LoRA, MoE, and KV cache management.* Design and implement compiler and runtime optimizations tailored for transformer-based models running on constrained, real-time platforms.* Collaborate with teams across CUDA, kernel libraries, compilers, and robotics to deliver high-performance, production-ready solutions.* Contribute to CUDA kernel and operator development for critical transformer components such as attention, GEMM, and MoE.* Benchmark, profile, and optimize inference performance across diverse embedded and automotive environments.* Stay ahead of the rapidly evolving LLM/VLM ecosystem and bring emerging techniques into product-grade software.**What we need to see:*** BS, MS, PhD, or equivalent experience in Computer Science, Electrical/Computer Engineering, or a closely related field.* 4+ years of relevant software development experience.* Deep understanding of transformer models and inference optimization techniques (e.g., quantization, tensor parallelism, or memory-efficient scheduling).* Proficient programming ability with modern C++ (C++11/14/17 and beyond).* Familiarity with popular LLM frameworks and libraries such as TensorRT, TensorRT-LLM, vLLM, SGLang, MLC-LLM, or FlashInfer.* A track record of strong software design, execution, and collaboration across fields.**Ways to stand out from the crowd:*** Demonstrated development experience or open-source contributions to LLM inference frameworks and libraries, such as SGLang, vLLM, or FlashInfer.* Proficiency with CUDA, including efficient kernel development, performance profiling, and GPU architecture fundamentals.* Prior work on autoregressive LLM serving systems, including speculative decoding or KV cache management.* Familiarity with compiler infrastructure for large language model inference.* Exposure to robotics or embedded AI pipelines, including optimizing for low-latency, resource-constrained systems.NVIDIA is widely considered to be one of the technology world’s most desirable employers. We hire some of the most brilliant and forward-thinking people in the world. If you thrive on innovation, autonomy, and technical excellence, come join us to shape the future of edge AI.#LI-HybridYour 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 uses AI tools in its recruiting processes.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.
Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Senior Software Engineer, Machine Learning Inference
Senior Software Engineer, Machine Learning Inference

NVIDIA Corporation • Santa Clara (CA)

Hybrid
USD 152,000 - 288,000
Equity
Benefits
Senior AI-Native Systems Software Engineer, TensorRT
Senior AI-Native Systems Software Engineer, TensorRT

NVIDIA Corporation • Santa Clara (CA)

On-site
USD 152,000 - 288,000
Equity
Benefits
Hybrid work model
Senior Software Engineer, Deep Learning Inference - TensorRT
Senior Software Engineer, Deep Learning Inference - TensorRT

NVIDIA Corporation • Santa Clara (CA)

Hybrid
USD 152,000 - 288,000
Equity
Benefits
Senior Deep Learning Software Engineer, Inference
Senior Deep Learning Software Engineer, Inference

NVIDIA Corporation • Northern (KY)

Hybrid
USD 152,000 - 288,000
Senior Software Engineer, AI and DL Kernel Libraries
Senior Software Engineer, AI and DL Kernel Libraries

NVIDIA Corporation • Santa Clara (CA)

On-site
USD 184,000 - 288,000
Senior Software Engineer – TensorRT Edge-LLM
Senior Software Engineer – TensorRT Edge-LLM

NVIDIA • California (MO)

On-site
USD 152,000 - 242,000
Equity
Benefits
Senior AI Software Engineer, Kernel Libraries
Senior AI Software Engineer, Kernel Libraries

NVIDIA Corporation • Santa Clara (CA)

On-site
USD 184,000 - 288,000
Senior Software Engineer, Deep Learning Inference - Automotive Safety
Senior Software Engineer, Deep Learning Inference - Automotive Safety

NVIDIA Corporation • Santa Clara (CA)

Hybrid
USD 152,000 - 288,000
Senior AI Workflow Engineer
Senior AI Workflow Engineer

NVIDIA Corporation • Santa Clara (CA), Northern (KY)

Hybrid
USD 184,000 - 357,000
Equity compensation
Health insurance
Relocation support
Software Engineer, TensorRT Specialized Platforms - New College Grad 2025
Software Engineer, TensorRT Specialized Platforms - New College Grad 2025

NVIDIA Corporation • Santa Clara (CA)

On-site
USD 124,000 - 196,000
Equity
Benefits