About the job
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.
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)
Qualifications
Minimum
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)