Senior DL Simulation Engineer — MLIR & GPU Modeling

Socket.dev; Santa Clara (TX); On-site; USD 152,000 - 288,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA in Santa Clara is seeking a Deep Learning Software Engineer to help develop simulation infrastructure for AI workloads across GPU hardware. You will work on fast, scalable evaluation across NVIDIA compiler stacks, improve kernel code generation and graph optimization, and help model datacenter-scale workloads.

The role requires MLIR experience, strong C/C++ and Python skills, and a Master's or PhD. Equity and benefits are offered, with a base salary schedule and a pathway to Level 3 or

Qualifications

• Master's degree (or equivalent) in CS/CE or related STEM; PhD preferred.
• 3+ years in compiler optimization, architectural simulation, or related areas.
• Strong hands-on experience with MLIR and compiler infrastructure.
• Excellent C/C++ and Python programming skills with design, debugging, and testing.
• Strong communication and collaboration in a fast-paced, cross-functional environment.

Responsibilities

• Develop simulation backends enabling fast, scalable evaluation of AI workloads across NVIDIA compiler stacks.
• Improve compiler kernel code generation and graph optimization using 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.

Skills

C/C++ programming

Python programming

Compiler optimization

Architectural simulation

Team collaboration

Education

Master's degree in CS/CE or related STEM; PhD preferred

PhD preferred

Tools

MLIR

Job description

NVIDIA in Santa Clara is seeking a Deep Learning Software Engineer to help develop simulation infrastructure for AI workloads across GPU hardware. You will work on fast, scalable evaluation across NVIDIA compiler stacks, improve kernel code generation and graph optimization, and help model datacenter-scale workloads.

The role requires MLIR experience, strong C/C++ and Python skills, and a Master's or PhD. Equity and benefits are offered, with a base salary schedule and a pathway to Level 3 or