Performance Engineer - Deep Learning

7 Minutes 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

Applike Group - Senior Data Scientist (Recommendation Systems Expert) (f/m/d)

Applike Group

Hamburg, Hamburg, Germany (Hybrid)
6 Months ago
Canva - Senior Machine Learning Engineer - Photo AI

Canva

Vienna, Vienna, Austria (Remote)
3 Months ago
ByteDance - Architect - AML Engine

ByteDance

San Jose, California, United States (On-Site)
6 Months ago
Google - Software Engineer III, Generative AI, Google Workspace

Google

Sunnyvale, California, United States (On-Site)
1 Week ago
ByteDance - Research Scientist (Machine Learning for Science (AI-for-Science))

ByteDance

Seattle, Washington, United States (On-Site)
4 Weeks ago
Hedra - Research Scientist

Hedra

San Francisco, California, United States (On-Site)
4 Weeks ago
Google - Software Engineer III, Embedded Systems/Firmware, Platforms Infrastructure Engineering

Google

Sunnyvale, California, United States (On-Site)
1 Week ago
Hedra - Applied Research Scientist

Hedra

San Francisco, California, United States (On-Site)
4 Weeks ago
NVIDIA - Software Engineering Intern, AI Engineering - 2025

NVIDIA

Shanghai, Shanghai, China (On-Site)
3 Months ago
Hedra - Research Engineer

Hedra

San Francisco, California, United States (On-Site)
4 Weeks ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Canva - Senior Applied Scientist - AI Research

Canva

Surry Hills, New South Wales, Australia (Remote)
1 Month ago
Krafton  - Applied Research Engineer - Reinforcement Learning

Krafton

Seoul, South Korea (On-Site)
4 Weeks ago
ByteDance - Applied Scientist Intern (Computational Modeling & Optimization)

ByteDance

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

Evolution

Warsaw, Masovian Voivodeship, Poland (Hybrid)
8 Months ago
Google - Solution Engineer, Innovation, Cloud Solution Accelerator Workshops

Google

New York, New York, United States (On-Site)
1 Week ago
Starkflow - Principal Full Stack Developer

Starkflow

Karnataka, India (Hybrid)
1 Month ago
Virtuos - Senior Games Tool Engineer (Machine Learning Specialist)

Virtuos

Shanghai, Shanghai, China (On-Site)
1 Month ago
Google - Software Engineer III, Machine Learning, Search

Google

Seattle, Washington, United States (On-Site)
5 Months ago
Rackspace Technology - Machine Learning Architect (AWS)

Rackspace Technology

San Diego, California, United States (Remote)
4 Weeks ago
Ubisoft - Scientifique en données ML Senior _ Groupe Technologique Content Creation

Ubisoft

Montreal, Quebec, Canada (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Santa Clara, California, United States

ByteDance - Software Engineer in ML Engineering Platform

ByteDance

Seattle, Washington, United States (On-Site)
5 Months ago
InMobiInMobi - Sales Manager

InMobiInMobi

New York, New York, United States (On-Site)
5 Months ago
PlayStation Global - Staff Machine Learning Engineer, Enterprise Enablement

PlayStation Global

Carlsbad, California, United States (On-Site)
4 Weeks ago
Kaedim - Customer Support Engineer

Kaedim

San Francisco, California, United States (On-Site)
8 Months ago
Samsung Semiconductor - Intern, Machine Learning Research Scientist

Samsung Semiconductor

San Jose, California, United States (Hybrid)
3 Months ago
The Walt Disney Company - Sign Painter

The Walt Disney Company

Anaheim, California, United States (On-Site)
1 Week ago
Visual Concepts - Senior Server Engineer

Visual Concepts

Novato, California, United States (On-Site)
7 Months ago
Google - Software Engineer III, Full Stack, Google Cloud Platforms

Google

Sunnyvale, California, United States (On-Site)
1 Week ago
Echtra Games - Senior Gameplay Engineer

Echtra Games

San Francisco, California, United States (On-Site)
9 Months ago
PlayStation Global - Senior Data Scientist

PlayStation Global

Carlsbad, California, United States (Remote)
1 Week ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Keywords Studios - AI - Research Associate (Prompts)

Keywords Studios

Silesian Voivodeship, Poland (On-Site)
1 Month ago
ByteDance - Research Scientist, Responsible AI

ByteDance

San Jose, California, United States (On-Site)
1 Month ago
ByteDance - Research Engineer (Foundation Model) - Machine Learning Systems

ByteDance

Singapore (On-Site)
5 Months ago
HP - AI Lab - Junior Machine Learning Engineer

HP

Sant Cugat Del Vallès, Catalonia, Spain (On-Site)
1 Month ago
NVIDIA - Senior AI-HPC Storage Engineer

NVIDIA

Westford, Massachusetts, United States (On-Site)
1 Month ago
NVIDIA - Senior Software Engineer, PyTorch - Deep Learning

NVIDIA

Santa Clara, California, United States (Hybrid)
3 Weeks ago
Bragg - Principal AI/ML Engineer

Bragg

Ljubljana, Ljubljana, Slovenia (Hybrid)
1 Week ago
ASSIST Software - AI Engineer

ASSIST Software

Suceava, Suceava County, Romania (Remote)
5 Months ago
Google - Developer Relations Engineer, AI Developer, Cloud AI

Google

Sunnyvale, California, United States (On-Site)
6 Days ago
NVIDIA - Solution Architect - Auto

NVIDIA

Shanghai, Shanghai, China (On-Site)
1 Month 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