Software Engineer, Pytorch - Deep Learning

NVIDIA is hiring software engineers at all experience levels to define and build the tools Deep Learning engineers use across the world to craft, 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 as well as directly contributes to PyTorch, a premiere Deep Learning framework. In this role you will work with multiple teams at NVIDIA across fields, as well as collaborate internationally with the PyTorch community to develop the best AI platform in the world!

What You Will Be Doing

• Design, build and support production AI solutions used by enterprise customers and partners.
• Work with internal applied researchers to improve their AI tools.
• Design and build core PyTorch components.
• Collaborate with NVIDIA’s hardware and software teams to improve GPU performance in PyTorch.
• Advise design of new hardware generations.

What We Need To See

• Proven experience developing large software projects.
• 6+ years of experience with C++ and Python programming.
• BS Computer Science, Computer Architecture, Electrical Engineering or a related field (or equivalent experience).
• We require excellent verbal and written communication skills.

Ways To Stand Out From The Crowd

• We participate in the open source community.
• Experience with parallel programming for CPU or GPU architectures.
• Experience in PyTorch.
• Proven experience working with multidisciplinary teams.

The base salary range is $176,000 - $333,500. 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 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