AI Computing Software Development Engineer, TensorRT

4 Months ago • 3 Years +

Job Summary

Job Description

NVIDIA's AI Computing team seeks a TensorRT Software Development Engineer to craft and develop robust inferencing software scalable across multiple platforms. Responsibilities include performance analysis, optimization, and tuning; tracking academic AI advancements and updating TensorRT; providing feedback on architecture and hardware; collaborating across teams to guide machine learning inferencing; and publishing results. The ideal candidate possesses a Master's degree (or equivalent experience) in a relevant field, 3+ years of software development experience, excellent C/C++ skills, deep learning framework experience (TensorFlow, PyTorch), and strong communication skills. The role involves working on a fast-paced team contributing to state-of-the-art AI and compute systems.
Must have:
  • Master's degree in relevant field
  • 3+ years software development experience
  • Excellent C/C++ programming skills
  • Deep learning framework experience (TensorFlow, PyTorch)
  • Strong communication skills
Perks:
  • Competitive salary
  • Generous benefits package

Job Details

We are now looking for a TensorRT Software Development Engineer!

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

  • 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:

  • Master or higher degree in Computer Engineering, Computer Science, Applied Mathematics or related computing focused degree (or equivalent experience)

  • 3+ years of 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.

With competitive salaries and a generous benefits package, NVIDIA is widely considered to be one of the most desirable employers in the world. We have some of the most brilliant and talented people in the world working for us. If you are creative, autonomous and love a challenge, we want to hear from you. We are an equal opportunity employer and value diversity at our company. We do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status.

Similar Jobs

NVIDIA - Software Engineering Manager, Sparse Linear Algebra Libraries

NVIDIA

United States (Hybrid)
4 Months ago
Single Store - AI Security Engineer

Single Store

Bengaluru, Karnataka, India (Remote)
1 Month ago
Flexra Software - Machine Learning Lead

Flexra Software

Tel Aviv-Yafo, Tel Aviv District, Israel (Hybrid)
1 Month ago
PayPal - Staff Machine Learning Engineer

PayPal

Dublin, County Dublin, Ireland (Hybrid)
1 Month ago
Amazon games  - Senior ML Scientist, Amazon Games AI Research

Amazon games

San Diego, California, United States (On-Site)
6 Months ago
Airlab Inc  - Junior Programmer Artificial Intelligence

Airlab Inc

Quebec, Canada (On-Site)
3 Months ago
Penn Interactive - Engineering Manager, ML Platform

Penn Interactive

Philadelphia, Pennsylvania, United States (Hybrid)
4 Months ago
Vigaet - Internship- AI Engineer

Vigaet

Bengaluru, Karnataka, India (On-Site)
1 Year ago
Tencent - Senior Researcher: Artificial General Intelligence (Natural Language Processing)

Tencent

Washington, United States (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

upwork - Lead Machine Learning Engineer - AI Agents

upwork

(Remote)
1 Month ago
Evolution  - Data Scientist

Evolution

Warsaw, Masovian Voivodeship, Poland (Hybrid)
11 Months ago
Hedra - Machine Learning Engineer (CUDA)

Hedra

San Francisco, California, United States (On-Site)
3 Months ago
Reddit - Senior Machine Learning Engineer

Reddit

(Remote)
3 Months ago
Hedra - Applied Research Scientist

Hedra

New York, New York, United States (On-Site)
3 Months ago
Google - Technical Program Manager III, AI/ML, Cloud AI Systems

Google

Austin, Texas, United States (On-Site)
2 Months ago
Canva - Senior Machine Learning Engineer - Photo AI

Canva

Vienna, Vienna, Austria (Remote)
5 Months ago
NVIDIA - Senior Site Reliability Engineer - AI Research Clusters

NVIDIA

Westford, Massachusetts, United States (Hybrid)
4 Months ago
bytedance - Research Scientist, Reinforcement Learning

bytedance

Seattle, Washington, United States (On-Site)
8 Months ago
Meta - Research Scientist Intern, Machine Perception for Input and Interaction (PhD)

Meta

Pittsburgh, Pennsylvania, United States (On-Site)
8 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Hsinchu, Hsinchu City, Taiwan

rivos - CPU Design Verification - Full-time

rivos

Hsinchu, Hsinchu City, Taiwan (Hybrid)
8 Months ago
Garena - Web Graphic Designer

Garena

Taipei City, Taiwan (On-Site)
2 Months ago
Corsair - HR Specialist

Corsair

Taiwan (On-Site)
2 Months ago
Google - Bluetooth Firmware Engineer

Google

New Taipei, New Taipei City, Taiwan (On-Site)
2 Months ago
appier - Software Engineer, QA Automation

appier

Taipei City, Taiwan (On-Site)
1 Month ago
Google - Test Engineer, Pixel Software Security

Google

New Taipei, New Taipei City, Taiwan (On-Site)
2 Months ago
NVIDIA - Research Scientist, Deep Learning and Computer Vision

NVIDIA

Taipei City, Taiwan (On-Site)
5 Months ago
rivos - Data Parallel Accelerator Performance Intern

rivos

Hsinchu, Hsinchu City, Taiwan (Hybrid)
8 Months ago
appier - Ad Sales, Advertising Solution

appier

Taipei City, Taiwan (On-Site)
1 Month ago
NXP - <2025 Internship Program> Application Engineer

NXP

Taipei City, Taiwan (On-Site)
8 Months ago

Get notifed when new similar jobs are uploaded

Similar Category Jobs

Lucid Reality Labs - Machine Learning Engineer

Lucid Reality Labs

Poland (Remote)
5 Months ago
Meta - Software Engineer, Machine Learning

Meta

Fremont, California, United States (Remote)
7 Months ago
NVIDIA - Principal Engineer

NVIDIA

(Remote)
4 Months ago
GoTo Group - Lead Data Scientist - KYC

GoTo Group

Singapore (On-Site)
6 Months ago
Meta - Software Engineer, Systems ML - SW/HW Co-design

Meta

Redmond, Washington, United States (On-Site)
7 Months ago
Scale AI - Machine Learning Engineer, International Public Sector

Scale AI

United Kingdom (On-Site)
8 Months ago
Microsoft - Member of Technical Staff, AI Post-Training

Microsoft

London, England, United Kingdom (On-Site)
3 Months ago
AI Fund - Head of AI @ Olakai

AI Fund

California, United States (Remote)
8 Months ago
Scale AI - QA Engineer, Generative AI

Scale AI

Argentina (On-Site)
8 Months ago
Google - Senior Imaging and On-Device Machine Learning Software Engineer, Silicon

Google

New Taipei, New Taipei City, Taiwan (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

About The Company

Since its founding in 1993, NVIDIA (NASDAQ: NVDA) has been a pioneer in accelerated computing. The company’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined computer graphics, ignited the era of modern AI and is fueling the creation of the metaverse. NVIDIA is now a full-stack computing company with data-center-scale offerings that are reshaping industry.

Santa Clara, California, United States (On-Site)

Massachusetts, United States (On-Site)

Santa Clara, California, United States (On-Site)

Texas, United States (On-Site)

Santa Clara, California, United States (Hybrid)

Santa Clara, California, United States (Hybrid)

Pune, Maharashtra, India (On-Site)

Taipei City, Taiwan (On-Site)

Beijing, Beijing, China (On-Site)

Santa Clara, California, United States (On-Site)

View All Jobs

Get notified when new jobs are added by NVIDIA

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug