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
Water / Wastewater Engineering Manager 6 - US Hybrid
Water / Wastewater Engineering Manager 6 - US Hybrid Date: Jan 23, 2024 Location: US Company: Black & Veatch Family of Companies Together, we own our company, our future, and our shared success.
Dispatcher
Dispatcher Location19 Natalie Way Plymouth, Massachusetts 02360 USPhone NumberCategoriesOperation SupportReq IDJR1372 Dispatcher (Open) First for a reasonFirst Student is the largest school transport
Phlebotomist
Description Want to Expand your career-development potential, your ability to help donors and patients, and your access to professional opportunities? We’re growing fast. [You can, too!] There are so
Intermediate Compliance Specialist (Hybrid Work Option)
36718BR Requisition ID: 36718BR Business Unit: COR Job Description: CDM Smith is seeking an Intermediate Compliance Specialist to join our Corporate Compliance Team. This individual performs basic to
Creative Lead / Designer II
Company Summary DISH, an EchoStar Company, is a Fortune 250 that is reimagining the future of connectivity. For over 40 years, we’ve been challenging the status quo and evolving our company to antici
Senior Member of Technical Staff (JoinOCI-SDE)
Job Description We are seeking a strong engineer to join our team which is focused on building and maintaining a scalable software control platform for Compute Infrastructure. Major focus areas of so
Health Technician (Dietetic)
Summary The Dayton Veterans Affairs Medical Center (VAMC) is recruiting for a Health Technician. The Health Technician will function with in Nutrition and Food Services. The primary purpose of the po
Hotel Laundry Attendant
Req ID: 431582 Address: 7101 W. Sundust Rd Chandler, AZ, 85226 Welcome to Love’s! * * Where People are the Heart of Our Success * * Hotel Laundry Attendant Laundry Attendants are expected to maintain
L2 Customer Technical Support Analyst - MICROS - Simphony/Ebusiness
Job Description L2 Customer Technical Support Analyst - MICROS - Simphony/Ebusiness Location: Orlando, FL or Columbia, MD highly preferred No visa sponsorship is available for this position. As a mem
Pharmacy Technician
Bring your heart to CVS Health. Every one of us at CVS Health shares a single, clear purpose: Bringing our heart to every moment of your health. This purpose guides our commitment to deliver enhanced
Copyright 2023-2025 - www.zdrecruit.com All Rights Reserved