Senior Software Development Engineer, TensorRT-LLM jobs in United States
cer-icon
Apply on Employer Site
company-logo

NVIDIA · 1 day ago

Senior Software Development Engineer, TensorRT-LLM

NVIDIA is a leading technology company focused on deep learning-powered AI. The role involves developing robust inferencing software for LLMs, collaborating across teams, and improving the usability of the TensorRT-LLM library.

AI InfrastructureArtificial Intelligence (AI)Consumer ElectronicsFoundational AIGPUHardwareSoftwareVirtual Reality
check
Growth Opportunities
check
H1B Sponsor Likelynote

Responsibilities

Craft and develop robust inferencing software that can be scaled to multiple platforms for functionality and performance
Perform benchmarking, profiling, and system-level programming for GPU applications
Closely follow academic developments in the field of artificial intelligence and feature update TensorRT
Provide code reviews, design docs, and tutorials to facilitate collaboration among the team
Conduct unit tests and performance tests for different stages of the inference pipeline
Collaborate across the company to guide the direction of machine learning inferencing, working with software, research and product teams
Write safe, scalable, modular, and high-quality (Python) code for our core backend software for LLM inference
Improve the usability of the TensorRT-LLM library and build systems (CMake)

Qualification

PythonDeep learning frameworksGPU programmingPerformance modelingSoftware designCuriosity about AICommunicationSelf-starter

Required

Masters or higher degree in Computer Engineering, Computer Science, Applied Mathematics or related computing focused degree (or equivalent experience)
4+ years of relevant software development experience
Excellent Python programming and software design skills, including debugging, performance analysis, and test design
Strong curiosity about artificial intelligence, awareness of the latest developments in deep learning like LLMs, generative and recommender models
Experience working with deep learning frameworks like TensorFlow and PyTorch
Self-starter who consistently takes initiative to drive projects forward
Excellent written and oral communication skills in English

Preferred

Prior experience with a LLM framework or a DL compiler in inference, deployment, algorithms, or implementation
Prior experience with performance modeling, profiling, debug, and code optimization of a DL/HPC/high-performance application
Architectural knowledge of CPU and GPU
GPU programming experience (CUDA or OpenCL)

Benefits

Equity
Benefits

Company

NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.

H1B Sponsorship

NVIDIA has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1877)
2024 (1355)
2023 (976)
2022 (835)
2021 (601)
2020 (529)

Funding

Current Stage
Public Company
Total Funding
$4.09B
Key Investors
ARPA-EARK Investment ManagementSoftBank Vision Fund
2023-05-09Grant· $5M
2022-08-09Post Ipo Equity· $65M
2021-02-18Post Ipo Equity

Leadership Team

leader-logo
Jensen Huang
Founder and CEO
linkedin
leader-logo
Michael Kagan
Chief Technology Officer
linkedin
Company data provided by crunchbase