Senior Deep Learning Software Engineer, DLSim

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

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

Ways to stand out from the crowd:

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

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 152,000 USD - 241,500 USD for Level 3, and 184,000 USD - 287,500 USD for Level 4.

We are now looking for a Deep Learning Software Engineer to help develop simulation infrastructure. The software rapidly assesses new AI-accelerating GPU hardware and software advancements., NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us. If you’re creative, collaborative and love a challenge, we want to hear from you!