Senior Deep Learning Software Engineer, DLSim

NVIDIA•Austin, TX

8d•$152,000 - $287,500

About The Position

The DL Performance Modeling Team develops 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 the strategic roadmap for deep learning hardware and software. This role develops simulation infrastructure that rapidly assesses advances in AI-accelerating GPU hardware and software.

Requirements

Master’s degree or equivalent experience in Computer Science, Computer Engineering, or a related STEM field.
At least 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.

Nice To Haves

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

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 modeled-scenario analysis 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.