Performance Engineer - Deep Learning

Company: NVIDIA
Location: Santa Clara
Posted on: November 14, 2024

Job Description:

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 - 230,000 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.

Keywords: NVIDIA, Sunnyvale , Performance Engineer - Deep Learning, Engineering , Santa Clara, California