Senior Deep Learning Architect

at Nvidia
USD 224,000-356,500 per year
SENIOR
βœ… On-site

SCRAPED

Used Tools & Technologies

Not specified

Required Skills & Competences ?

Python @ 4 Algorithms @ 4 Machine Learning @ 4 System Architecture @ 4 PyTorch @ 3 CUDA @ 3 GPU @ 3

Details

We are seeking an experienced engineer to help design the next generation of accelerators and system architecture to enable AI performance and efficiency improvements. The role offers the opportunity to directly impact future hardware and system roadmaps at a leading company in AI computing.

Responsibilities

  • Understand, analyze, profile, and optimize AI training workloads on state-of-the-art hardware and software platforms.
  • Guide development of future generations of artificial intelligence accelerators and systems.
  • Develop detailed performance models and simulator infrastructure for computing systems accelerating AI training, and implement and evaluate hardware feature proposals.
  • Collaborate across teams (hardware, software, research, production) to guide the direction of machine learning at the company.
  • Drive HW/SW co-design of the full deep learning platform stack, from silicon to deep learning frameworks.

Requirements

  • PhD in Computer Science, Electrical Engineering, or related field with 5+ years of experience; or MS (or equivalent experience) with 8+ years of relevant work experience.
  • Strong background in computer architecture, with a proven track record of architecting features in shipping high-performance processors.
  • Background in artificial intelligence and large language models, particularly training algorithms and workloads.
  • Experience analyzing and tuning application performance on state-of-the-art hardware.
  • Experience with processor and system-level performance modelling, simulation, and evaluation prior to silicon availability.
  • Programming skills in C++ and Python.
  • Familiarity with GPU computing across all layers of the AI stack, from deep learning frameworks (e.g., PyTorch) down to CUDA.

Benefits

  • Base salary range: 224,000 USD - 356,500 USD (determined based on location, experience, and comparable roles).
  • Eligible for equity and company benefits.
  • Applications accepted through at least July 29, 2025.

Location

  • Santa Clara, CA, United States

Additional Information

  • Employer highlights: NVIDIA is described as a leader in GPUs and AI computing, committed to diversity and equal opportunity.

#deeplearning