Performance Engineer - Deep Learning

3 Months 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 worldwide. The role involves optimizing model performance for NVIDIA GPUs, tuning large-scale training workloads, optimizing production AI models for enterprise clients, contributing to benchmarks (MLPerf), and collaborating with the open-source community. Responsibilities include enhancing the performance of cutting-edge models, leveraging new hardware and software innovations, and informing the design of future hardware and software components. The ideal candidate will have experience with C++, Python, parallel programming (GPUs), and a strong understanding of computer architecture or operating systems.
Must have:
  • C++ and Python programming
  • Parallel programming (GPUs)
  • Knowledge of Computer Architecture/OS
  • Experience with large software projects
  • Excellent communication skills
Good to have:
  • PyTorch, TensorFlow, or MXNet experience
  • Performance analysis and profiling
  • Open-source community participation
  • Multidisciplinary team experience
Perks:
  • Highly competitive salaries
  • 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

Canva - Senior Machine Learning Engineer - Photo AI

Canva

Prague, Czechia (Remote)
2 Months ago
PlayStation Global - Staff Machine Learning Engineer, Enterprise Enablement

PlayStation Global

California, United States (On-Site)
3 Months ago
Framestore - Machine Learning Developer - London Launchpad Internship 2025

Framestore

England, United Kingdom (On-Site)
1 Month ago
NVIDIA - Solutions Architect, Financial Services

NVIDIA

New York, New York, United States (Remote)
3 Months ago
NVIDIA - Senior Software Engineer - Robot Learning Platform

NVIDIA

Toronto, Ontario, Canada (On-Site)
1 Month ago
DraftKings - Lead Machine Learning Engineer, Platform

DraftKings

Boston, Massachusetts, United States (On-Site)
2 Months ago
Zoox - Senior Software Engineer: Secure Embedded Operating Systems

Zoox

Foster City, California, United States (On-Site)
6 Months ago
Birdeye - Senior Product Manager

Birdeye

Gurugram, Haryana, India (On-Site)
6 Months ago
Zoox - Manager, Simulation Traffic & Behavior Modeling

Zoox

Foster City, California, United States (Hybrid)
6 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Epic Games - Principal Research Scientist

Epic Games

Pittsburgh, Pennsylvania, United States (On-Site)
3 Months ago
Samsung Semiconductor - Staff Engineer, AI/ML Software Compiler

Samsung Semiconductor

San Jose, California, United States (Hybrid)
4 Months ago
ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - Speech & Audio) - 2025 Start (PhD)

ByteDance

Seattle, Washington, United States (On-Site)
5 Months ago
Dream Sports - Senior ML Scientist

Dream Sports

Mumbai, Maharashtra, India (On-Site)
5 Months ago
NVIDIA - Deep Learning Performance Architect

NVIDIA

Beijing, Beijing, China (On-Site)
3 Months ago
PwC - Associate

PwC

Bengaluru, Karnataka, India (On-Site)
5 Months ago
NVIDIA - Senior DevOps Engineer, Deep Learning Frameworks

NVIDIA

Warsaw, Masovian Voivodeship, Poland (On-Site)
3 Months ago
Meta - Research Scientist Intern, Machine Perception for Input and Interaction (PhD)

Meta

Seattle, Washington, United States (On-Site)
5 Months ago
Visa - Senior Manager Data Science - Visa Consulting & Analytics

Visa

Mumbai, Maharashtra, India (On-Site)
6 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Santa Clara, California, United States

Logitech - Global eCommerce Business Engagement Lead

Logitech

United States (Remote)
4 Months ago
PlayStation Global - SDET Contractor

PlayStation Global

Aliso Viejo, California, United States (On-Site)
2 Months ago
NVIDIA - Material Planner

NVIDIA

Santa Clara, California, United States (On-Site)
3 Months ago
The Walt Disney Company - Community Engagement & Development Director

The Walt Disney Company

Houston, Texas, United States (On-Site)
2 Months ago
The Walt Disney Company - Manager, Product Design (Disney +)

The Walt Disney Company

Seattle, Washington, United States (On-Site)
5 Months ago
Zoox - Senior/Staff Motion Planning Engineer, Teleguidance

Zoox

Foster City, California, United States (Hybrid)
6 Months ago
Trackman - Trackman Range Customer Success Manager

Trackman

Arizona, United States (On-Site)
2 Months ago
Anavation - Vulnerability Researcher

Anavation

Chantilly, Virginia, United States (On-Site)
6 Months ago
Aristocrat Gaming - Lead Data Engineer - ED&I

Aristocrat Gaming

Las Vegas, Nevada, United States (Hybrid)
4 Months ago
PENN Interactive - Engineering Manager, ML Platform

PENN Interactive

Philadelphia, Pennsylvania, United States (Hybrid)
2 Months ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Like Card - Senior AI Research Scientist

Like Card

Amman, Amman Governorate, Jordan (On-Site)
1 Month ago
The Walt Disney Company - Principal Software Engineer

The Walt Disney Company

Burbank, California, United States (On-Site)
3 Months ago
Imagineio - Senior Generative AI Engineer

Imagineio

India (Remote)
5 Months ago
Interface AI - Senior Technical Recruiter

Interface AI

United States (Remote)
2 Months ago
Globalization Partners - Principal AI Engineer

Globalization Partners

(Remote)
3 Months ago
Match Group - Senior ML Platform Engineer

Match Group

New York, New York, United States (Hybrid)
6 Months ago
Zoox - Software Engineer - 3D World Generation Pipelines

Zoox

Seattle, Washington, United States (Hybrid)
6 Months ago
Inworld AI - Staff / Principal Machine Learning Engineer - USA

Inworld AI

Mountain View, California, United States (Remote)
5 Months ago
Meta - Software Engineer, Machine Learning

Meta

Pittsburgh, Pennsylvania, United States (On-Site)
5 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.

California, United States (Remote)

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

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)

Bengaluru, Karnataka, India (On-Site)

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

Pune, Maharashtra, India (On-Site)

Taipei City, Taiwan (On-Site)

Taipei City, Taiwan (On-Site)

Beijing, Beijing, China (On-Site)

Tel Aviv-Yafo, Tel Aviv District, Israel (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