Senior Deep Learning Software Engineer - Developer Tools

$148,000 - $276,000/Yr

Nvidia - Westford, MA

Full-time - Mid Level

Westford, MA

Computer and Electronic Product Manufacturing

About the position

NVIDIA is seeking an outstanding software engineer to join the Developer Tools organization, which is dedicated to delivering best-in-class tools for developers. This role focuses on innovating tools that support the design and performance optimizations of deep neural network models within a unified environment. As part of this team, you will partner with management and architects to translate requirements into actionable designs and lead the development process. Staying up to date with the latest research in deep learning is crucial, as you will be expected to become a domain authority in this field. Your contributions will directly impact the efficiency and effectiveness of deep learning applications, making this a pivotal role in the organization.

Responsibilities

• Innovate on tools support for design and performance optimizations of deep neural network models.
• Partner with management and architects to translate requirements into designs and lead development.
• Stay up to date with the latest research and become a domain authority in deep learning.

Requirements

• Master's degree in computer science, mathematics, or related engineering fields (or equivalent experience).
• At least 5 years of relevant technical work experience.
• Excellent knowledge of deep learning and hands-on experience working with computer vision and large language models.
• Working experience with DL frameworks such as PyTorch, Tensorflow, or JAX.
• Strong programming skills in Python or C++.
• Excellent communication and interpersonal skills.
• Self-motivated and ability to work with geographically distributed and multi-functional teams.

Nice-to-haves

• In-depth knowledge of PyTorch and ONNX.
• Familiar with NVIDIA DL software stacks (TensorRT, etc.).
• Hands-on experience with developing deep learning applications, from training to deployment.
• Experience and/or interests in building software tools for designing and optimizing deep neural network models.

Benefits

• Equity and benefits package based on location and experience.