Senior Deep Learning Software Engineer, DLSim

at Nvidia
USD 152,000-287,500 per year
SENIOR
✅ On-site

Tech Stack

AI @ 4 Communication @ 7 Debugging @ 6 Deep Learning GPU LLM @ 7 Performance Analysis @ 6 Python @ 6

Details

The DL Performance Modeling Team’s core mission is to deliver full-stack simulation infrastructure for deep learning applications across a spectrum of GPUs. The team collaborates with architecture, software, product, and research teams to shape and refine the strategic roadmap for deep learning hardware and software.

The role focuses on developing simulation infrastructure that rapidly assesses new AI-accelerating GPU hardware and software advancements.

Responsibilities

  • Develop simulation backends that enable fast, scalable evaluation of AI workloads across NVIDIA compiler stacks.
  • Improve deep learning compiler kernel code generation and computational graph optimization using analysis based on modeled scenarios and performance insights.
  • Advance the modeling and optimization of datacenter-scale AI workloads and deployment scenarios.
  • Partner with architects and software teams to evaluate future GPU features and guide silicon and system-level design decisions.

Requirements

  • Master’s degree, or equivalent experience, in Computer Science, Computer Engineering, or a related STEM field; PhD preferred.
  • 3+ years of relevant experience in compiler optimization, architectural simulation, or related areas.
  • Strong hands-on experience with MLIR and compiler infrastructure.
  • Excellent C/C++ and Python programming skills, including software design, debugging, performance analysis, and test development.
  • Strong communication and collaboration skills, with the ability to thrive in a fast-paced, multifunctional, outcome-focused environment.

Preferred Qualifications

  • Experience designing and building compiler frameworks or intermediate representations from the ground up.
  • Deep understanding of LLM inference workloads and their implications for computer architecture.
  • Hands-on experience implementing and optimizing complex AI workloads on CPUs, GPUs, or custom accelerators.

Benefits

  • Equity and benefits are provided.
  • NVIDIA is committed to fostering an inclusive work environment and is an equal opportunity employer.

Additional Information

  • This is a full-time position.
  • Applications will be accepted at least until August 28, 2026.
  • The posting is for an existing vacancy.
  • NVIDIA uses AI tools in its recruiting processes.
  • Base salary depends on location, experience, and the pay of employees in similar positions. The base salary range is USD 152,000–241,500 for Level 3 and USD 184,000–287,500 for Level 4.

More jobs at Nvidia

Similar jobs