Senior Deep Learning Performance Architect - LPU
$152k - $288k • US, CA, Remote
Posted 1d ago
Job Location
US, CA, Remote
Tech Stack
Remote Work Policy
On-site
Categories
Machine Learning Engineer
About the job
NVIDIA seeks a Senior DL Performance Architect to join a pioneering group focused on pushing AI Inference performance boundaries through ambitious hardware-software co-design. This role offers an outstanding opportunity to develop world-class performance strategies, guide future GPU architecture decisions, and lead AI innovation. If you are passionate about AI efficiency, have a proven record of modeling LLM performance and architecting AI systems, and enjoy optimizing every cycle, this role may be perfect for you.
Responsibilities
- Design novel GPU and system architectures to advance AI Inference performance and efficiency.
- Construct, investigate, and test popular deep learning algorithms and applications.
- Analyze the relationship between hardware and software architectures influencing future algorithms and applications.
- Build efficient power and performance models of the AI inference stack to guide next-gen HW architecture.
- Collaborate across teams to guide the direction of AI development.
Requirements
- MS or PhD in a relevant field (CS, EE, Math) or equivalent experience.
- 5+ years of relevant experience.
- Strong mathematical foundation in machine learning and deep learning.
- Expert programming skills in C, C++, and/or Python.
- Familiarity with GPU computing (CUDA or similar) and HPC (MPI, OpenMP) stack.
- Strong knowledge and coursework in computer architecture.
Benefits
- Equity
- Benefits