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 cutting-edge architectures for AI and high-performance computing. Responsibilities include performance analysis, modeling, and developing analytical models, simulators, and test suites. The role involves understanding hardware/software interplay, utilizing deep learning frameworks, and collaborating with software, product, and research teams. This position requires strong expertise in GPU or Deep Learning ASIC architecture, performance modeling, machine learning, and programming in Python, C, and C++. The ideal candidate will have experience with frameworks like PyTorch, JAX, and TensorRT.
Must have:
  • 6+ years experience
  • GPU/DL ASIC architecture expertise
  • Performance modeling & analysis
  • Machine learning & deep learning knowledge
  • Python, C, C++ programming skills
Good to have:
  • PyTorch, JAX, TensorRT experience
  • CUDNN, CUBLAS, CUTLASS, MLIR, Triton, CUDA, OpenCL experience
  • DL accelerator architecture or workload analysis
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

GoMotive - Software Engineer, Machine Learning

GoMotive

Pakistan (Remote)
2 Months ago
Flip Fit - Machine Learning Engineer

Flip Fit

(Remote)
4 Months ago
Google - Software Engineer III, Ads

Google

Mountain View, California, United States (On-Site)
1 Month ago
ByteDance - Research Scientist, Foundation Model, Speech & Audio

ByteDance

San Jose, California, United States (On-Site)
7 Months ago
NVIDIA - Senior Tool and Methodology Development Software Engineer

NVIDIA

Taipei City, Taiwan (On-Site)
4 Months ago
Meta - AI Research Scientist, Language - Generative AI

Meta

Redmond, Washington, United States (On-Site)
6 Months ago
Google - Software Engineer III, AI/ML, Google Play

Google

Mountain View, California, United States (On-Site)
1 Month ago
Meta - Software Engineer, Machine Learning

Meta

San Francisco, California, United States (On-Site)
7 Months ago
Google - Senior Software Engineer, AI/ML, Google Ads

Google

Mountain View, California, United States (On-Site)
1 Month ago
Google - Senior Software Engineer, SDLC, Gemini Code Assist

Google

Kirkland, Washington, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

ByteDance - Backend Software Engineer, Data Engineering

ByteDance

Singapore (On-Site)
7 Months 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
Wargaming - Game Developer

Wargaming

Belgrade, Serbia (On-Site)
2 Months ago
2K - Applied Scientist

2K

Dublin, County Dublin, Ireland (Hybrid)
1 Month ago
N-iX - 2D/3D Visualization Engineer

N-iX

Ukraine (Remote)
4 Months ago
Google - Software Engineer III, Google Distributed Cloud Hosted

Google

Sunnyvale, California, United States (On-Site)
1 Month ago
Google - Senior Software Engineer, Infrastructure, Google Cloud Security and Privacy

Google

Cambridge, Massachusetts, United States (On-Site)
1 Month ago
ByteDance - Research Scientist Graduate (Foundation Model - Vision and Language)

ByteDance

Seattle, Washington, United States (On-Site)
4 Months ago
Tradu - Data Scientist

Tradu

Chennai, Tamil Nadu, India (On-Site)
8 Months ago
ION - Senior AI Engineer, Italy

ION

Pisa, Tuscany, Italy (On-Site)
7 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Redmond, Washington, United States

Infosys - Lead .NET Full Stack Developer

Infosys

Hartford, Connecticut, United States (On-Site)
1 Month ago
Anavation LLC - Senior Vulnerability Researcher

Anavation LLC

Lorton, Virginia, United States (Hybrid)
2 Months ago
Neolytix - Credentialing Specialist

Neolytix

Chicago, Illinois, United States (Hybrid)
4 Weeks ago
Google - Technical Program Manager II, Pixel Carrier Engineering, Devices and Services

Google

Mountain View, California, United States (On-Site)
1 Month ago
ByteDance - Research Scientist in Foundation Model, Speech Understanding - 2025 Start (PhD)

ByteDance

San Jose, California, United States (On-Site)
7 Months ago
Absurd Ventures - Senior/Lead Gameplay Engineer

Absurd Ventures

Santa Monica, California, United States (On-Site)
1 Month ago
Google - Research Scientist, Google Cloud AI

Google

Kirkland, Washington, United States (On-Site)
1 Month ago
ByteDance - Video Analysis and Quality Algorithm Intern 2023 Summer/Fall (PHD)

ByteDance

San Diego, California, United States (On-Site)
7 Months ago
Interactive Brokers - Bilingual Trade Issues (Spanish/English Fluent)

Interactive Brokers

Fort Lauderdale, Florida, United States (On-Site)
1 Month ago
Netflix - Data Scientist (L5) - Product Promotion & Algorithm Performance

Netflix

Los Angeles, California, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Globalization Partners - Senior AI Engineer

Globalization Partners

(Remote)
3 Months ago
ASSIST Software - AI Engineer

ASSIST Software

Suceava, Suceava County, Romania (Remote)
6 Months ago
Airlab Inc  - Junior Programmer Artificial Intelligence

Airlab Inc

Quebec, Canada (On-Site)
2 Months ago
Google - Staff Software Engineer, ML Performance, GPUs

Google

Sunnyvale, California, United States (On-Site)
1 Month ago
AI Fund - General Manager - New Business Unit (College Admissions)

AI Fund

California, United States (Remote)
7 Months ago
Google - Software Engineer III, Generative AI

Google

Sunnyvale, California, United States (On-Site)
1 Month ago
Egnyte - Senior Product Manager

Egnyte

Mountain View, California, United States (Remote)
6 Months ago
ByteDance - Researcher - Large Language Models, Applied Machine Learning

ByteDance

Seattle, Washington, United States (On-Site)
2 Months ago
Microsoft - Senior Data Scientist

Microsoft

(On-Site)
1 Month ago
Meta - Software Engineer, Machine Learning

Meta

New York, New York, United States (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