Home
/
Comprehensive
/
Software Engineer, Torch-TensorRT Inference
Software Engineer, Torch-TensorRT Inference-March 2024
Santa Clara
Mar 17, 2025
ABOUT NVIDIA
NVIDIA is a computing platform company, innovating at the intersection of graphics, HPC, and AI.
10,000+ employees
Technology
VIEW COMPANY PROFILE >>
About Software Engineer, Torch-TensorRT Inference

  We are looking for outstanding Software Engineers for Torch-TensorRT Inference framework! As a member of our Torch-TensorRT Engineering team, you will use groundbreaking NVIDIA deep learning model inference software libraries for deployment on NVIDIA's hardware architecture. You will develop a deep learning compiler and optimize DNN graphs in lower precision using NVIDIA TensorRT!

  We work on innovative vision and language models to ensure they are accelerated by Torch-TensorRT. In addition, we contribute to our open-source library and have opportunities to present our work at conferences.

  What you will be doing:

  Develop components of Torch-TensorRT, NVIDIA's SDK for high-performance deep learning inference along with PyTorch

  Use C++, Python and CUDA to build graph parsers, converters, compiler, and tools for effective deployment of trained deep learning models.

  Work with cross-collaborative teams of deep learning software engineers and GPU architects to innovate across applications like autonomous driving, NLP, and computer vision.

  What we need to see:

  A BS, MS or PhD in Computer Architecture, Computer Science, Electrical Engineering or related field or equivalent experience

  5+ years of work related experience in software development or Machine learning or high-performance computing

  2+ years of experience in developing or using deep learning frameworks (e.g., PyTorch, TensorFlow, ONNX, etc.)

  Strong C++ programming skills including debugging, performance analysis, test design, and optimizations.

  Strong understanding of build/release systems, CI/CD, and experience with solutions like Gitlab, Github, etc.

  Ways to stand out from the crowd:

  Familiarity with sophisticated C++11/C++14 language features and CUDA kernel programming.

  Experience working in an open-source and software development on Windows.

  Strong time-management and organization skills for coordinating multiple initiatives, priorities and implementations of new technology and products into very complex projects.

  The base salary range is 116,000 USD - 270,250 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

  You will also be eligible for equity and benefits (https://www.nvidia.com/en-us/benefits/) . NVIDIA accepts applications on an ongoing basis.

  NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.

  NVIDIA is a Learning Machine

  NVIDIA pioneered accelerated computing to tackle challenges no one else can solve. Our work in AI and the metaverse is transforming the world's largest industries and profoundly impacting society.

  Learn more about NVIDIA .

Comments
Welcome to zdrecruit comments! Please keep conversations courteous and on-topic. To fosterproductive and respectful conversations, you may see comments from our Community Managers.
Sign up to post
Sort by
Show More Comments
SIMILAR JOBS
Field Sales Representative
Job Description: Our Field Sales Representatives earn between $43,100.00 - $64,700.00 + up to $20,000 in commissions if all sales goals are met. With our uncapped commission opportunities, overachiev
Senior Associate/Manager_Java/Scala Developer_Technology_ FRPPE
Morgan StanleyJava Developer - PFTCompany Profile Morgan Stanley is a leading global financial services firm providing a wide range of investment banking, securities, wealth management and investment
Teamleiter (w/m/d) Technical Project- and Property Management
Job Family: Real Estate Req ID: 401979 Zur Verstärkung unseres Teams bei Innomotics suchen wir einen Teamleiter (w/m/d) Technical Project- and Property Management . Ihr Antrieb zur Veränderung. Motor
Casino Marketing Coordinator
Overview The incumbent in this position is primarily responsible for performing highly skilled administrative duties and assisting with the execution of the casino marketing calendar. The Marketing C
Sr Sales Operation Analyst
Job Description Sales Operation Analyst of PSG revenue operations under Business Management Group manages the sales operation process, performs forecasting, planning, and analysis activities of Intel
Grants Writer - Infectious Diseases and Rheumatology
Reference #: 1006765SHIFT:Day (United States of America) Seeking Breakthrough Makers Children's Hospital of Philadelphia (CHOP) offers countless ways to change lives. Our diverse community of more th
Helpdesk Technician
The helpdesk technician will Provide technical assistance and troubleshooting related to personal computer (PC) software & hardware and telecommunication services, including connectivity, to all
Controller
Reference #: R2199Are you ready to link your passion with a purpose? At NantHealth, we build technology that simplifies healthcare. We manage the data and decisions that help the healthcare industry
Diesel Mechanic Technician
Diesel Mechanic Technician needed.  Must be at least 21 y/o and have at least 3 years experience.  Preferred that you have your own tools and reliable transportation.  Must have experience with scan
CNC Milling Machinist-Experienced (1st Shift, Weekend, Fri-Mon, Part-Time/Full Time)
JOB SUMMARY The position of an experienced CNC Milling Machinist Technician is designed for a knowledgeable and skilled individual in multiple fields of machining. This position is responsible for th
Copyright 2023-2025 - www.zdrecruit.com All Rights Reserved