AI Computing Development Intern, TensorRT-LLM - 2027
At a glance
Junior AI Engineering role at NVIDIA. China, Shanghai · full-time.
Pay not stated AI Engineering salary
aimlroles summary, based on the employer's posting.
What you'll do
- Build inference software that can scale across platforms, with attention to both functionality and speed
- Analyze performance, then optimize and tune TensorRT-LLM
- Track AI research and use relevant developments to update TensorRT-LLM features
- Share input on architecture and hardware design, coordinating with software, research and product teams
- Present important results at scientific conferences
What you bring
- You are pursuing a master’s degree or higher in computer engineering, computer science, applied mathematics or a related computing field
- You have relevant software development experience and strong C/C++ skills, including debugging, performance analysis and test design
- You have used TensorFlow or PyTorch and follow advances in deep learning, including LLMs, generative models or recommender systems
- You communicate effectively in written and spoken English
Who this fits
This internship suits a graduate student with software development experience who can work independently on a delivery-focused team. You'll collaborate across software, research and product groups.
From the employer
We are now looking for a TensorRT-LLM Software Development Engineering intern!
NVIDIA is hiring software engineers for its AI Computing 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 GenerativeAI that has put DL at the “iPhone moment” for AI. Join the team which is building the inferencing software which will be used across our product lines! The ability to work on a fast-paced delivery-focused team is required and excellent interpersonal skills are a must.
What you'll be doing:
Craft and develop robust inferencing software that can be scaled to multiple platforms for functionality and performance
Performance analysis, optimization and tuning
Closely follow academic developments in the field of artificial intelligence and feature update TensorRT-LLM
Provide feedback into the architecture and hardware design and development
Collaborate across the company to guide the direction of machine learning inferencing, working with software, research and product teams
Publish key results in scientific conferences
What we need to see:
Pursuing a Masters or higher degree in Computer Engineering, Computer Science, Applied Mathematics or related computing focused degree
Relevant software development experience.
Excellent C/C++ 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
Proactive and able to work without supervision
Excellent written and oral communication skills in English
NVIDIA is widely considered to be one of technology’s most desirable employers. We have some of the most forward-thinking and hardworking people on the planet working for us. Does the idea of contributing to and pushing the boundaries of state-of-the-art AI and Compute systems excite you? Interested in getting exposure to the entire DL SW stack? Come join us and help build the GPU-accelerated DL platform used worldwide.