Senior Deep Learning Software Engineer, DLSim
$152k - $288k • US, CA, Santa Clara
Posted 1mo ago
Job Location
US, CA, Santa Clara
Tech Stack
Remote Work Policy
On-site
Categories
Machine Learning Engineer
About the job
The DL Performance Modeling Team is seeking a Deep Learning Software Engineer to develop full-stack simulation infrastructure for deep learning applications across various GPUs. This role involves collaborating with cross-functional teams to shape the roadmap for DL hardware and software, and to rapidly assess new AI-accelerating GPU hardware and software advancements through simulation.
Responsibilities
- Develop simulation backends for fast, scalable evaluation of AI workloads on NVIDIA compiler stacks.
- Improve deep learning compiler kernel code generation and computational graph optimization.
- 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 design decisions.
Requirements
- Master's degree (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.
Benefits
- Equity
- Benefits