Senior Performance Engineer - LLM Inference Frameworks
You will design and optimize high‑performance inference pipelines for large language models on NVIDIA GPUs, focusing on throughput, memory efficiency, and scalability. Your work will directly shape the TensorRT‑LLM framework, enabling state‑of‑the‑art LLM inference across the AI community.
Skills required
About this role
NVIDIA is hiring exceptional software engineers to build and optimize the core inference infrastructure for large language models. Join the TensorRT‑LLM team - the group defining how generative AI performs at global scale on NVIDIA GPUs. We’re looking for engineers who love squeezing every drop of throughput, memory efficiency, and scalability out of modern model runtimes. Your work will directly shape the frameworks behind state‑of‑the‑art LLM inference used across NVIDIA and the AI community. Join us to redefine what “fast” means for LLM inference - building the frameworks that power the next generation of generative AI at scale.
What you'll be doing:
Design, implement, and optimize high‑performance inference pipelines for large language models running on GPUs
Profile and tune model execution across the stack - from scheduler design to kernel fusions and everything in-between
Design and experiment with memory management strategies for improved memory bandwidth optimization and cache efficiency
Innovate and Implement cutting-edge techniques such as Speculative Decoding, Context Caching, and FP8/INT4 quantization to push the boundaries of tokens-per-second-per-watt
Develop and maintain benchmarking and testing systems that quantify latency, utilization, and efficiency
What we need to see:
Bachelor's, Master's, or higher degree in Computer Engineering, Computer Science, Applied Mathematics, or related computing-focused degree (or equivalent experience)
5+ years of relevant software development experience.
Excellent Python programming skills, software design, and software engineering skills
Experience working with deep learning frameworks like PyTorch and HuggingFace
Experience profiling and debugging performance at all levels - Python runtime, PyTorch internals, and GPU utilization metrics
Awareness of the latest developments in LLM architectures and LLM inference techniques
Proactive and able to work without supervision
Excellent written and oral communication skills in English
Ways to stand out from the crowd:
Contributions to inference frameworks such as TensorRT‑LLM, vLLM, SGLang, or similar systems
Demonstrated expertise in performance modeling, memory optimization, distributed model execution or GPU execution workflows
Hands‑on experience with NVIDIA profiling tools (Nsight Systems, PyTorch Profiler, custom benchmarking harnesses)
Strong grasp of the trade‑offs shaping inference efficiency: compute vs. memory, scheduling vs. batching, latency vs. throughput
Widely considered to be one of the technology world’s most desirable employers, NVIDIA offers highly competitive salaries and a comprehensive benefits package. As you plan your future, see what we can offer to you and your family www.nvidiabenefits.com.
#LI-Hybrid
