NVIDIA Logo

NVIDIA

Senior Software Engineer – TensorRT Edge-LLM

Reposted 24 Days Ago
In-Office or Remote
2 Locations
152K-288K Annually
Senior level
In-Office or Remote
2 Locations
152K-288K Annually
Senior level
The role involves developing an inference framework for large language models and optimizing performance on embedded platforms, collaborating across teams and contributing to CUDA kernel development.
The summary above was generated by AI

Are 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-Hybrid

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 2, 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.

HQ

NVIDIA Santa Clara, California, USA Office

2701 San Tomas Expressway, Santa Clara, CA, United States, Santa Clara

NVIDIA San Francisco, California, USA Office

San Francisco, United States

NVIDIA San Jose, California, USA Office

San Jose, United States

Similar Jobs

36 Minutes Ago
Remote or Hybrid
USA
110K-160K Annually
Mid level
110K-160K Annually
Mid level
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
Manage a team of Technical Account Managers, enhancing customer success through strategic leadership, collaboration, and effective case resolution in the cybersecurity sector.
Top Skills: Ai-Native PlatformCybersecurityEndpoint Protection Technologies
36 Minutes Ago
Remote or Hybrid
USA
160K-250K Annually
Senior level
160K-250K Annually
Senior level
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
Lead design and build of an AI-native marketing platform: integrations, multi-agent orchestration, UIs, tooling, analytics, CI/CD, auth/multi-tenancy, and production AI agents to automate marketing workflows and drive adoption across teams.
Top Skills: AWSCi/CdClaudeClaude Code AgentsCmsContainerizationCrewaiDigital Asset ManagementEmbeddingsGoogle DocsGptHtml EmailJavaScriptLangchainLlm ApisMarketoMcp ServersOauthPdfPptxPythonRagReactRest ApisSlackTypescriptVector DatabasesWebhooksWorkfront
36 Minutes Ago
Remote or Hybrid
USA
115K-160K Annually
Senior level
115K-160K Annually
Senior level
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
Lead planning, deployment, tuning, and operational support of CrowdStrike Falcon's AI/ML detection and response capabilities. Implement and troubleshoot AI-driven detection models, automation playbooks, API integrations, and threat-hunting for customers. Provide training, mentorship, and customer-facing collateral while optimizing detection efficacy and integrating Falcon into customers' security workflows.
Top Skills: Active DirectoryAttack Surface ManagementAWSAzureAzure AdBashBatchChefCitrixCrowdstrike FalconData ProtectionFalcon ApisFile Integrity ManagementGCPGroup PolicyHost-Based FirewallIbm BigfixLogscaleMachine LearningMcafee EpoMS OfficeMssqlNlpPdqPowershellPsexecPuppetPythonRSccmSIEMSplunkSsoVulnerability Management

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account