Senior Deep Learning Performance Architect

at Nvidia
📍 Santa Clara, United States
USD 184,000-356,500 per year
SENIOR
✅ On-site

SCRAPED

Used Tools & Technologies

Not specified

Required Skills & Competences ?

Python @ 7 Algorithms @ 4 Machine Learning @ 4 PyTorch @ 4 CUDA @ 4 GPU @ 7

Details

We are now seeking a Senior Deep Learning Performance Architect!

NVIDIA is looking for outstanding Performance Architects with a background in performance analysis, performance modeling, and AI/deep learning to help analyze and develop the next generation of architectures that accelerate AI and high-performance computing applications.

Responsibilities

  • Develop innovative architectures to extend the state of the art in deep learning performance and efficiency
  • Analyze performance, cost and power trade-offs by developing analytical models, simulators and test suites
  • Understand and analyze the interplay of hardware and software architectures on future algorithms, programming models and applications
  • Develop, analyze, and harness groundbreaking Deep Learning frameworks, libraries, and compilers
  • Actively collaborate with software, product and research teams to guide the direction of deep learning hardware and software

Requirements

  • MS or PhD in Computer Science, Computer Engineering, Electrical Engineering or equivalent experience
  • 6+ years of meaningful work experience
  • Strong background in GPU or Deep Learning ASIC architecture for training and/or inference
  • Experience with performance modeling, architecture simulation, profiling, and analysis
  • Solid foundation in machine learning and deep learning
  • Strong programming skills in Python, C, C++

Ways to stand out

  • Background with deep neural network training, inference and optimization in leading frameworks (e.g. PyTorch, JAX, TensorRT)
  • Experience with relevant libraries, compilers, and languages - cuDNN, cuBLAS, CUTLASS, MLIR, Triton, CUDA, OpenCL
  • Experience with the architecture of or workload analysis on other deep learning accelerators
  • Demonstration of self-motivation, critical thinking and thinking outside the box

Compensation & Benefits

  • Base salary range: 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5
  • You will also be eligible for equity and benefits (see NVIDIA benefits page)

Additional information

  • Applications for this job will be accepted at least until October 28, 2025.
  • NVIDIA is committed to fostering a diverse work environment and is an equal opportunity employer.