Performance Engineer - Deep Learning

1 Week ago • 2 Years + • Artificial Intelligence • $120,000 PA - $235,750 PA

Job Summary

Job Description

NVIDIA seeks software engineers to optimize deep learning tools used globally for AI application design, development, and deployment. The role involves optimizing model performance for NVIDIA GPUs, tuning large-scale training workloads, optimizing production AI models for enterprise clients, contributing to community benchmarks (like MLPerf), and working with cutting-edge models. Collaboration with internal teams and the open-source community is crucial. The position influences NVIDIA's AI platform and contributes directly to frameworks like TensorFlow, PyTorch, and MXNet. It also involves informing the design of new hardware and software components.
Must have:
  • C++ and Python programming
  • Parallel programming (GPU experience)
  • Computer architecture/OS knowledge
  • Large software project development
  • Excellent communication skills
Good to have:
  • PyTorch, TensorFlow, or MXNet experience
  • Performance analysis and profiling
  • Open-source community participation
  • Multidisciplinary team experience
Perks:
  • Competitive salary
  • Comprehensive benefits package
  • Equity

Job Details

NVIDIA is hiring software engineers at all experience levels to build and optimize the tools Deep Learning engineers use across the world to design, develop, and deploy AI applications. This position will embed you in an ambitious and diverse team that influences all areas of NVIDIA's AI platform and directly contributes to premiere Deep Learning frameworks - Tensorflow, PyTorch, and MXNet. In this role you will work with multiple teams at NVIDIA across fields, as well as collaborate with the open-source community to optimize the best AI platform in the world.

What you will be doing:

  • Optimize the performance of Deep Learning models for NVIDIA GPUs and systems.

  • Study and tune Deep Learning training workloads at large scale.

  • Optimize production AI models used by enterprise customers and partners.

  • Build and support NVIDIA submissions to community benchmarks like MLPerf.

  • Optimize the performance of influential, contemporary models coming out of academic and industry research, for NVIDIA GPUs and systems.

  • Deliver the benefits of NVIDIA’s latest hardware and platform software innovations to the Deep Learning community.

  • Inform design of new hardware generations, and core platform software components for NVIDIA GPUs and systems.

What we need to see:

  • BS or equivalent experience in Computer Science, Electrical Engineering or a related field.

  • 2+ years of experience with C++ and Python programming.

  • Strong background with parallel programming, preferably on GPUs.

  • Knowledge of Computer Architecture and/or Operating Systems.

  • Proven experience developing large software projects.

  • Excellent verbal and written communication skills.

Ways to stand out from the crowd:

  • Experience in PyTorch, Tensorflow or MXNet.

  • Background with performance analysis and profiling of workloads.

  • Participation in the open source community.

  • Proven experience working with multidisciplinary teams.

With highly competitive salaries and a comprehensive benefits package, NVIDIA is widely considered to be one of the technology industry's most desirable employers. We have some of the most forward-thinking and dedicated people in the world working with us and our engineering teams are contributing to some of the hottest state of the art fields: Deep Learning, Artificial Intelligence, and Autonomous Vehicles. If you're a creative and motivated software engineer with a real passion for building fast software solutions, as well as impacting AI development worldwide, we want to hear from you.

The base salary range is 120,000 USD - 235,750 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.

#deeplearning

Similar Jobs

Google - Customer Engineer II, AI/ML

Google

Chicago, Illinois, United States (On-Site)
2 Days ago
Nintendo - Intern – Machine Learning Software Engineer (NTD)

Nintendo

Redmond, Washington, United States (On-Site)
5 Months ago
IMC - Machine Learning Engineer

IMC

Chicago, Illinois, United States (On-Site)
17 Hours ago
Meta - Research Scientist Intern, Machine Perception for Input and Interaction (PhD)

Meta

Burlingame, California, United States (On-Site)
5 Months ago
SmileGate - Game Data Engineer [LOST ARK]

SmileGate

Seongnam-si, Gyeonggi-do, South Korea (On-Site)
3 Months ago
PlayStation Global - Senior Machine Learning Engineer

PlayStation Global

London, England, United Kingdom (On-Site)
1 Week ago
Sony Interactive Entertainment - AI Learning and Development Coordinator

Sony Interactive Entertainment

Tokyo, Japan (On-Site)
3 Months ago
NVIDIA - Research Scientist, Deep Learning and Computer Vision

NVIDIA

Taipei City, Taiwan (On-Site)
3 Months ago
AI Fund - Head of Engineering

AI Fund

United States (Remote)
3 Weeks ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Krafton  - Applied Research Engineer - Reinforcement Learning

Krafton

Seoul, South Korea (On-Site)
1 Month ago
Attentive - Staff Software Engineer, Cloud FinOps

Attentive

(Remote)
3 Weeks ago
Reality Games - Machine Learning Engineer - Monopoly World

Reality Games

Kraków, Lesser Poland Voivodeship, Poland (On-Site)
2 Months ago
Electronic Arts - Senior Machine Learning Engineer

Electronic Arts

Hyderabad, Telangana, India (On-Site)
1 Month ago
Attentive - Senior Machine Learning Engineer

Attentive

San Francisco, California, United States (Hybrid)
6 Months ago
ByteDance - Software Engineer, ML System Scheduling

ByteDance

San Jose, California, United States (On-Site)
6 Months ago
ByteDance - Machine Learning Engineer - Inference

ByteDance

San Jose, California, United States (On-Site)
1 Month ago
Ciklum - Expert Data Scientist

Ciklum

Pune, Maharashtra, India (Hybrid)
6 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Santa Clara, California, United States

Falcon X - Risk Manager, Liquidity and Credit

Falcon X

New York, New York, United States (On-Site)
1 Day ago
ByteDance - Algorithm Engineer Intern (Multimedia Lab - Video Processing and Enhancement)

ByteDance

San Jose, California, United States (On-Site)
2 Months ago
Epic Games - Director, Marketing

Epic Games

Cary, North Carolina, United States (On-Site)
2 Weeks ago
Patreon - Scientist to Data Scientist

Patreon

San Francisco, California, United States (Hybrid)
4 Weeks ago
Google - Senior Software Engineer, Machine Learning (Recommendations, Rankings, and Predictions)

Google

Mountain View, California, United States (On-Site)
2 Days ago
Google - Customer Engineer, Infrastructure Modernization, Google Cloud

Google

San Francisco, California, United States (On-Site)
2 Weeks ago
Snail Games - Bilingual Translator & Executive Assistant (English/Mandarin)

Snail Games

Beverly Hills, California, United States (On-Site)
5 Months ago
Rackspace Technology - Client Partner V - Midwest

Rackspace Technology

United States (Remote)
2 Weeks ago
Betson Group - Engineering Tech lead

Betson Group

Malta, New York, United States (On-Site)
16 Hours ago
The Walt Disney Company - Lead Software Engineer, Ad Platforms

The Walt Disney Company

San Francisco, California, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Trend Micro - Sr. AI Engineer

Trend Micro

Taipei City, Taiwan (On-Site)
7 Months ago
Lionbridge Games - Language AI (Games) Program Manager

Lionbridge Games

Masovian Voivodeship, Poland (On-Site)
2 Months ago
Google - Software Engineer, Runtime, AICore, Platforms and Devices

Google

Taipei City, Taiwan (On-Site)
2 Days ago
ByteDance - Research Scientist Graduate (Foundation Model, Video Generation) - 2025 Start (PhD)

ByteDance

San Jose, California, United States (On-Site)
6 Months ago
NVIDIA - Deep Learning Intern - Fall 2025

NVIDIA

Shanghai, Shanghai, China (On-Site)
2 Weeks ago
Ello - Tech Lead, Generative AI & Machine Learning

Ello

San Francisco, California, United States (On-Site)
1 Month ago
Electronic Arts - Associate, Corporate Strategy

Electronic Arts

Redwood City, California, United States (Hybrid)
3 Months ago
Microsoft - Member of Technical Staff, AI Multimodal

Microsoft

London, England, United Kingdom (On-Site)
1 Month ago
NVIDIA - Senior Developer Relations Manager - Robotics

NVIDIA

Tokyo, Japan (On-Site)
3 Months ago
N-iX - Senior Data Scientist/AI Engineer

N-iX

Poland (Hybrid)
2 Weeks 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)

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

Santa Clara, California, United States (Hybrid)

Santa Clara, California, United States (Hybrid)

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