Senior Deep Learning Software Engineer, DLSim

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 insight.
• 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.
• What we need to see
• A Masters (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, multi-functional, outcome-focused environment.

Languages