Performance Engineer - Deep Learning

Posted 18 Days Ago
Be an Early Applicant
Santa Clara, CA
Mid level
Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
The Role
As a Performance Engineer, you'll optimize Deep Learning tools and workloads for NVIDIA GPUs, collaborating across teams and with the open-source community.
Summary Generated by Built In

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

Top Skills

C++
Mxnet
Python
PyTorch
TensorFlow
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Santa Clara, CA
21,960 Employees
On-site Workplace
Year Founded: 1993

What We Do

NVIDIA’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning ignited modern AI — the next era of computing — with the GPU acting as the brain of computers, robots, and self-driving cars that can perceive and understand the world. Today, NVIDIA is increasingly known as “the AI computing company.”

Similar Jobs

NVIDIA Logo NVIDIA

Senior Performance Engineer - Deep Learning

Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
Remote
2 Locations
21960 Employees

NVIDIA Logo NVIDIA

Performance Engineer - Deep Learning

Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
Santa Clara, CA, USA
21960 Employees

NVIDIA Logo NVIDIA

Senior Performance Engineer - Deep Learning

Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
Santa Clara, CA, USA
21960 Employees

Similar Companies Hiring

True Anomaly Thumbnail
Software • Machine Learning • Hardware • Defense • Artificial Intelligence • Aerospace
Colorado Springs, CO
131 Employees
Caliola Engineering Thumbnail
Software • Machine Learning • Hardware • Defense • Data Privacy • App development • Aerospace
Colorado Springs, CO
53 Employees
Red 6 Thumbnail
Virtual Reality • Software • Hardware • Defense • Aerospace
Orlando, Florida
113 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account