Senior Deep Learning Software Engineer, DLSim

What you'll do

• Develop simulation backends for scalable evaluation of AI workloads across NVIDIA compiler stacks.
• Improve compiler kernel code generation and computational graph optimization using modeled scenarios and performance insights.
• Model and optimize datacenter-scale AI workloads, and collaborate with architects and software teams on future GPU features and design decisions.

What they ask for

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

Nice to have

• PhD.
• Experience building compiler frameworks or intermediate representations, understanding LLM inference workloads and architecture implications, or implementing and optimizing complex AI workloads on CPUs, GPUs, or custom accelerators.

Pay

152,000 USD - 241,500 USD base salary (Level 3); 184,000 USD - 287,500 USD base salary (Level 4)

Tools

MLIR, C/C++, Python, NVIDIA compiler stacks

AI focus

Build simulation infrastructure to evaluate AI workloads and GPU advancements; optimize deep learning compiler code generation and graph optimization; model datacenter-scale AI workloads.