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 . 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.

#J-18808-Ljbffr