Deep Learning Performance Architect

4 Months ago • 2 Years + • Artificial Intelligence

Job Summary

Job Description

As a Deep Learning Performance Architect at NVIDIA, you'll benchmark and analyze AI workloads, develop simulators and debuggers in C++/Python, evaluate performance, power, and area (PPA) trade-offs, and collaborate with architecture teams. Responsibilities include analyzing hardware features, keeping abreast of deep learning trends, and communicating technical concepts effectively. You'll contribute to building real-time, cost-effective computing platforms for AI applications.
Must have:
  • MS/PhD in relevant field (CS, EE, Math)
  • 2+ years experience in parallel computing
  • Strong programming skills (C, C++, Python)
  • Proficiency in architecture analysis & modeling
  • Excellent problem-solving skills
Good to have:
  • Understanding of transformer-based models
  • Experience with benchmarking and workload profiling

Job Details

NVIDIA has continuously reinvented itself. Our invention of the GPU sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. Today, research in artificial intelligence is booming worldwide, which calls for highly scalable and massively parallel computation horsepower that NVIDIA GPUs excel.

NVIDIA is a “learning machine” that constantly evolves by adapting to new opportunities that are hard to solve, that only we can address, and that matter to the world. This is our life’s work , to amplify human creativity and intelligence. As an NVIDIAN, you’ll be immersed in a diverse, supportive environment where everyone is inspired to do their best work. Come join our diverse team and see how you can make a lasting impact on the world!

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, cost-effective computing platforms driving our success in this exciting and rapidly growing field!

What you'll be doing:

  • Benchmark and analyze AI workloads in single and multi-node configurations.

  • High level simulator and debugger development in C++/Python.

  • Evaluate PPA (performance, power, area) for hardware features and system-level architectural trade-offs.

  • Work closely with wider architecture teams, architecture and product management to help with trade-off analysis at every stage of the project.

  • Keep abreast with emerging trends and research in deep learning.

What we need to see:

  • MS or PhD in a relevant discipline (CS, EE, Math).

  • 2+ years of experience in parallel computing architectures, interconnect fabrics and deep learning applications.

  • Strong programming skills in C, C++ and Python.

  • Proficiency in architecture analysis and performance modeling.

  • Curious mindset with excellent problem solving skills.

Ways to stand out from the crowd: 

  • Understanding of modern transformer-based model architectures.

  • Experience with benchmarking, projections methodologies, workload profiling and correlation.

  • Ability to simplify and communicate rich technical concepts with non-technical audience.

#LI-Hybrid

Similar Jobs

Mozilla - Staff Machine Learning Engineer, Gen AI

Mozilla

Denmark (Remote)
7 Months ago
Numrah - Software Engineer (AI)

Numrah

(On-Site)
2 Months ago
NVIDIA - Machine Learning Intern - 2025

NVIDIA

(On-Site)
4 Months ago
Trend Micro - Sr. Data Scientist (AI Lab)

Trend Micro

Taipei City, Taiwan (On-Site)
8 Months ago
Fictiv - QA- SDET (Playwright)

Fictiv

Pune, Maharashtra, India (On-Site)
2 Weeks ago
Microsoft - Member of Technical Staff, AI Post-Training

Microsoft

Zürich, Zurich, Switzerland (On-Site)
1 Month ago
Meta - Software Engineer, Systems ML - SW/HW Co-design

Meta

New York, New York, United States (Remote)
6 Months ago
NVIDIA - Senior AI Training Performance Engineer

NVIDIA

Shanghai, Shanghai, China (Hybrid)
4 Months ago
PlayStation Global - Sr. ML Software Engineer

PlayStation Global

United States (Remote)
2 Months ago
CharacterAI - Research Engineer, Post-Training

CharacterAI

New York, New York, United States (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Nintendo - Sr Audio Algorithms Software Engineer (NTD)

Nintendo

Redmond, Washington, United States (On-Site)
10 Months ago
Zscaler - Staff Data Science Engineer

Zscaler

San Jose, California, United States (On-Site)
2 Weeks ago
Google - Staff Image Quality Evaluation Engineer, Silicon

Google

Mountain View, California, United States (On-Site)
1 Month ago
Opendoor - Software Engineering Manager, Growth

Opendoor

United States (Remote)
1 Week ago
conga - Sr. Software Engineer

conga

Bengaluru, Karnataka, India (On-Site)
1 Week ago
Adobe - Machine Learning Engineer

Adobe

San Jose, California, United States (On-Site)
6 Days ago
playrix  - UA Marketing Producer

playrix

Portugal (Remote)
7 Months ago
InMobiInMobi - Staff Data Scientist I

InMobiInMobi

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Inkittt - Senior Front-End Engineer: Web and Mobile Team (m/f/d)

Inkittt

Krakow Am See, Mecklenburg-Vorpommern, Germany (On-Site)
10 Months ago
NVIDIA - Senior AI-HPC Storage Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Bengaluru, Karnataka, India

Lytx - Staff Architect

Lytx

India (On-Site)
1 Month ago
Passion Gaming - Customer Service Executive (Kannada)

Passion Gaming

Panchkula, Haryana, India (On-Site)
1 Year ago
Survay Monkey - Senior Software Engineer I (QA)

Survay Monkey

Bengaluru, Karnataka, India (Hybrid)
1 Month ago
Accenture - Supply Chain Operations Analyst

Accenture

Mumbai, Maharashtra, India (On-Site)
1 Month ago
Qualcomm - Network Engineer

Qualcomm

Bengaluru, Karnataka, India (On-Site)
6 Days ago
P99 soft - Senior DevOps Engineer

P99 soft

Hyderabad, Telangana, India (On-Site)
1 Month ago
PwC - Senior Analyst - Deals - TS

PwC

Mumbai, Maharashtra, India (On-Site)
8 Months ago
Hitachi - Data Scientist

Hitachi

Pune, Maharashtra, India (Remote)
7 Months ago
Passion Gaming - Back End Engineer PHP

Passion Gaming

Gurugram, Haryana, India (On-Site)
9 Months ago
Workassist - Full Stack Dotnet Developer

Workassist

Pune, Maharashtra, India (On-Site)
8 Months ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

A-Team - AI Strategy Lead

A-Team

New York, New York, United States (Hybrid)
2 Months ago
bytedance - Research Scientist Intern (Doubao (Seed) - Machine Learning System) - 2025 Summer (PhD)

bytedance

San Jose, California, United States (On-Site)
7 Months ago
PlayStation Global - Senior Machine Learning Engineer

PlayStation Global

London, England, United Kingdom (On-Site)
1 Month ago
bytedance - Research Scientist, Vision Foundation Model

bytedance

San Jose, California, United States (On-Site)
7 Months ago
Meta - Software Engineer, Systems ML - SW/HW Co-design

Meta

Menlo Park, California, United States (Remote)
6 Months ago
NVIDIA - Principal Software Engineer - Enterprise AI Platform

NVIDIA

Santa Clara, California, United States (Hybrid)
4 Months ago
bytedance - LLM Software Engineer/Researcher (Applied Machine Learning)

bytedance

Seattle, Washington, United States (On-Site)
1 Month ago
Amazon games  - Senior ML Scientist, Amazon Games AI Research

Amazon games

San Diego, California, United States (On-Site)
5 Months ago
FTF Studios - FTF Senior Programmer

FTF Studios

(Remote)
1 Year ago
Scale AI - QA Engineer, Generative AI

Scale AI

Argentina (On-Site)
7 Months 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)

Massachusetts, United States (On-Site)

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

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

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

Texas, United States (On-Site)

Santa Clara, California, United States (Hybrid)

Santa Clara, California, United States (Hybrid)

Pune, Maharashtra, India (On-Site)

Taipei City, Taiwan (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