Senior Deep Learning Performance Architect

3 Months ago • 6 Years + • Artificial Intelligence • $184,000 PA - $356,500 PA

Job Summary

Job Description

NVIDIA seeks a Senior Deep Learning Performance Architect to develop innovative architectures for deep learning performance and efficiency. Responsibilities include analyzing performance trade-offs, developing analytical models and simulators, understanding hardware/software interplay, working with deep learning frameworks, and collaborating with software, product, and research teams. The ideal candidate will have a strong background in GPU or deep learning ASIC architecture, performance modeling, machine learning, and programming skills (Python, C, C++).
Must have:
  • MS or PhD in relevant field
  • 6+ years experience
  • GPU/DL ASIC architecture expertise
  • Performance modeling & analysis skills
  • Deep learning & ML foundation
  • Proficiency in Python, C, C++
Good to have:
  • Experience with PyTorch, JAX, TensorRT
  • Familiarity with CUDNN, CUBLAS, CUTLASS, MLIR, Triton, CUDA, OpenCL
  • Experience with other DL accelerators
Perks:
  • Equity
  • Benefits

Job 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.

What you’ll be doing:

  • 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 HW and SW

What we need to see:

  • 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 from the crowd:

  • 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 DL accelerators

  • Demonstration of self-motivation, with a knack for critical thinking and thinking outside the box

Intelligent machines powered by Artificial Intelligence computers that can learn, reason and interact with people are no longer science fiction. GPU Deep Learning has provided the foundation for machines to learn, perceive, reason and solve problems. NVIDIA's GPUs run AI algorithms, simulating human intelligence, and act as the brains of computers, robots and self-driving cars that can perceive and understand the world. Increasingly known as “the AI computing company”, NVIDIA wants you! Come, join our Deep Learning Architecture team, where you can help build real-time, efficient computing platforms driving our success in this exciting and rapidly growing field.

The base salary range is 184,000 USD - 356,500 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.

Similar Jobs

Microsoft - Research Intern - Data Systems

Microsoft

Redmond, Washington, United States (On-Site)
6 Days ago
Trackman - Machine Learning Developer with DSP experience (Python/C++)

Trackman

Hørsholm, Denmark (On-Site)
4 Weeks ago
Google - Lead CPU RTL Engineer, Silicon

Google

Poughkeepsie, New York, United States (On-Site)
4 Months ago
NVIDIA - Senior ASIC Verification Engineer - GPU Memory Subsystem

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago
Rackspace Technology - Principal MLOps Engineer

Rackspace Technology

San Antonio, Texas, United States (Remote)
4 Weeks ago
Google - Staff Software Engineer, AI/ML

Google

Hyderabad, Telangana, India (On-Site)
1 Week ago
NVIDIA - Senior Software Engineer, AI Resiliency

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago
Razer - Solutions Architect

Razer

Singapore (On-Site)
6 Months ago
Google - Software Engineer III, AI/ML

Google

Bengaluru, Karnataka, India (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Google - Software Engineer, Black Community Inclusion

Google

(On-Site)
5 Months ago
ByteDance - Research Scientist, Foundation Model, Speech Understanding

ByteDance

Seattle, Washington, United States (On-Site)
5 Months ago
Google - Software Engineer III, Google Store

Google

Mountain View, California, United States (On-Site)
6 Days ago
Google - Software Engineer III, Infrastructure, Platforms Infrastructure Engineering

Google

Sunnyvale, California, United States (On-Site)
4 Months ago
ByteDance - Software Engineer - MySQL

ByteDance

Seattle, Washington, United States (On-Site)
2 Months ago
PlayStation Global - Machine Learning Engineer for Game Technology

PlayStation Global

Aliso Viejo, California, United States (On-Site)
9 Months ago
Google - Software Engineer, Android Automotive

Google

Kraków, Lesser Poland Voivodeship, Poland (On-Site)
1 Week ago
Google - Staff Software Engineer

Google

Pune, Maharashtra, India (On-Site)
1 Week ago
Mozilla - Staff Machine Learning Engineer, Gen AI

Mozilla

Belgium (Remote)
6 Months ago
ByteDance - Research Scientist for Generative AI, Multimodal and LLM

ByteDance

San Jose, California, United States (On-Site)
1 Week ago

Get notifed when new similar jobs are uploaded

Jobs in Santa Clara, California, United States

Meta - Research Engineer, SysML - FAIR

Meta

Boston, Massachusetts, United States (Remote)
1 Week ago
The Walt Disney Company - Lead Software Engineer (Front End/JavaScript)

The Walt Disney Company

Glendale, California, United States (On-Site)
5 Months ago
Google - Field Sales Manager III, Healthcare, Google Cloud

Google

Chicago, Illinois, United States (On-Site)
1 Week ago
Netflix - UI Engineer (L5) - Content Engineering

Netflix

United States (Remote)
5 Months ago
Microsoft - Principal Technical Program Manager, Master Data Management and Governance

Microsoft

Redmond, Washington, United States (Hybrid)
4 Days ago
Google - Senior Software Engineer, Infrastructure, Google Cloud Storage

Google

New York, New York, United States (On-Site)
1 Week ago
Twitch - Software Engineer

Twitch

San Francisco, California, United States (On-Site)
1 Month ago
Google - Technical Program Manager I, Data Center Infrastructure

Google

Atlanta, Georgia, United States (On-Site)
1 Week ago
Epic Games - Game Designer, Season Pass

Epic Games

Cary, North Carolina, United States (On-Site)
1 Week ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Meta - Research Scientist Intern, Machine Perception for Input and Interaction (PhD)

Meta

Burlingame, California, United States (On-Site)
5 Months ago
Krafton  - Head of Deep Learning PM & Ops Dept.

Krafton

Seoul, South Korea (On-Site)
4 Weeks ago
Google - Software Engineer II, Applied AI

Google

Kraków, Lesser Poland Voivodeship, Poland (On-Site)
1 Week ago
Microsoft - Member of Technical Staff, AI Pre-Training

Microsoft

Zürich, Zurich, Switzerland (On-Site)
1 Week ago
QUANTIC DREAM - Programmeur Intelligence Artificielle

QUANTIC DREAM

Montreal, Quebec, Canada (Hybrid)
6 Months ago
Google - Senior Research Engineer, AI/ML

Google

London, England, United Kingdom (On-Site)
4 Days ago
Google - Staff Software Engineer, ML Infrastructure

Google

Sunnyvale, California, United States (On-Site)
1 Week ago
Level AI - Backend Engineer - Customer Engineering

Level AI

Noida, Uttar Pradesh, India (Hybrid)
6 Months ago
Tencent - Game AI Researcher

Tencent

Tokyo, Japan (On-Site)
3 Weeks ago
Google - Software Engineer III, Distributed Machine Learning, YouTube

Google

Mountain View, California, United States (On-Site)
4 Days ago

Get notifed when new similar jobs are uploaded

About The Company

Since its founding in 1993, NVIDIA (NASDAQ: NVDA) has been a pioneer in accelerated computing. The company’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined computer graphics, ignited the era of modern AI and is fueling the creation of the metaverse. NVIDIA is now a full-stack computing company with data-center-scale offerings that are reshaping industry.

Santa Clara, California, United States (On-Site)

Santa Clara, California, United States (On-Site)

Santa Clara, California, United States (On-Site)

Yokne'am Illit, North District, Israel (On-Site)

Yokne'am Illit, North District, Israel (On-Site)

Yokne'am Illit, North District, Israel (On-Site)

Yokne'am Illit, North District, Israel (On-Site)

Santa Clara, California, United States (On-Site)

Santa Clara, California, United States (On-Site)

View All Jobs

Get notified when new jobs are added by NVIDIA

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug