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
Material Handler I
JOB REQUIREMENTS: Company Generac Power Systems Name Material HandlerI - Weekend Day (Plant) Req # 63983 Employment Type Full Time ShiftWeekend Day The Material Handler I will be responsible for all
Team Member
Team Member We’re glad you’re here. Think about it – you and us? Sounds like it could be a match made in the meat heavens. If you’re here for: Discounted Curly Fries (and all our menu items for that
Phlebotomist
At Labcorp we have a passion in helping people live happy and healthy lives. Every day we provide vital information that helps our clients and patients understand their health. If you are passionate
Cook
Hot burgers, cold shakes, and little moments of magic right in the neighborhood. At SONIC, we do things a little differently. We find the fun, the moment of chill in the every-day. Working at SONIC,
Team Member
Team Member We’re glad you’re here. Think about it – you and us? Sounds like it could be a match made in the meat heavens. If you’re here for: Discounted Curly Fries (and all our menu items for that
P07647UF - Assistant Professor of Teaching - Music
Minimum/Required Qualifications Doctoral level degree OR terminal degreein Music or related discipline. Demonstrable knowledge of the musictechnology industry, current equipment, and professional pra
Lead Machine Learning Engineer
is seeking a lead engineer to work in the machine learning (ML) and artificial intelligence (AI) domains for our Syracuse, NY corporate headquarters. The rapid advancement of deep learning continues
Team Member
Team Member We’re glad you’re here. Think about it – you and us? Sounds like it could be a match made in the meat heavens. If you’re here for: Discounted Curly Fries (and all our menu items for that
Principal Member of Technical Staff
Job Description Building off our Cloud momentum, Oracle has formed a new organization - Oracle Health Applications & Infrastructure. This team focuses on product development and product strategy
Maintenance Technician II (Carpenter/Mason/Painter)
Maintenance Technician II (Carpenter/Mason/Painter) Apply now (https://secure.dc4.pageuppeople.com/apply/822/gateway/default.aspx?c=apply&lJobID=521795&lJobSourceTypeID=801&sLanguage=en-u
Copyright 2023-2025 - www.zdrecruit.com All Rights Reserved