NVIDIA

Senior AI Software Development Engineer, TensorRT-LLM

Posted: 1 minutes ago

Job Description

We are now looking for a TensorRT-LLM Software Development Engineer!NVIDIA is hiring software engineers for its TensorRT-LLM team. Academic and commercial groups around the world are using GPUs to power a revolution in deep learning-powered AI, enabling breakthroughs in areas like LLM, ChatGPT and Generative AI that have put DL at the “iPhone moment” for AI. Join the team which is building the inferencing software which is foundational to product lines within NVIDIA and across the industry! The ability to work on a fast-paced delivery-focused team is required and excellent interpersonal skills are a must.What You'll Be DoingCraft and develop robust inference software that can be scaled to multiple platforms for functionality and performancePerformance analysis, optimization, and tuning for Large Language Models (LLMs)Conduct unit tests and performance tests for different stages of the inference pipeline.Closely follow academic developments in the field of artificial intelligence and feature update TensorRT-LLMWrite safe, scalable, modular, and high-quality (C++/Python) code for our core backend software for LLM inference.Collaborate across the company to guide the direction of deep learning inference, working with software, research and product teamsWhat We Need To SeeBachelors, Masters 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 skillsAwareness of the latest developments in LLM architectures and LLM inference techniquesExperience working with deep learning frameworks like PyTorch and HuggingFaceProactive and able to work without supervisionExcellent written and oral communication skills in EnglishWays To Stand Out From The CrowdPrior experience with a LLM inference framework (TensorRT-LLM, SGLang, vLLM, etc.) or a DL compiler in inference, deployment, algorithms, or implementationPrior experience with performance modeling, profiling, debug, and code optimization of a DL/HPC/high-performance applicationExcellent C/C++ programming and software design skills, including debugging, performance analysis, and test design.Architectural knowledge of CPU and GPUGPU programming experience (CUDA or OpenCL)JR2008357

Job Application Tips

  • Tailor your resume to highlight relevant experience for this position
  • Write a compelling cover letter that addresses the specific requirements
  • Research the company culture and values before applying
  • Prepare examples of your work that demonstrate your skills
  • Follow up on your application after a reasonable time period

You May Also Be Interested In