Senior Software Engineer - Developer Tools for Deep Learning

at Nvidia
πŸ“ World
πŸ“ United States
USD 152,000-287,500 per year
SENIOR
βœ… Remote

Used Tools & Technologies

LLM

Required Skills & Competences

Python @ 7 TensorFlow @ 4 Communication @ 4 Mathematics @ 4 PyTorch @ 4 Deep Learning @ 4 AI @ 4 Computer Vision @ 7 TensorRT @ 3 JAX @ 4

Details

We are building developer tools to enable streamlined design and optimization of high-performance deep neural network models in a unified environment. Join the Developer Tools organization at NVIDIA to help deliver best-in-class tooling for deep learning practitioners and developers.

Responsibilities

  • Enhance tool support to improve deep neural network design and performance efficiency.
  • Partner with management and architects to translate requirements into designs and own the development.
  • Stay up to date with the latest research and become a domain expert in deep learning.

Requirements

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

Ways to Stand Out

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

Compensation & Benefits

  • Base salary ranges provided by level:
    • Level 3: 152,000 USD - 241,500 USD
    • Level 4: 184,000 USD - 287,500 USD
  • You will also be eligible for equity and benefits.

Additional Information

  • Applications accepted at least until June 9, 2026.
  • This posting is for an existing vacancy.
  • NVIDIA uses AI tools in its recruiting processes.
  • NVIDIA is an equal opportunity employer committed to fostering an inclusive work environment.