Software Engineer, PyTorch - Deep Learning



Software Engineering
Helsinki, Finland
Posted on Tuesday, January 9, 2024

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 and build core PyTorch components.

  • Collaborate with NVIDIA’s hardware and software teams to improve GPU performance in PyTorch.

  • Design, build and support production AI solutions used by enterprise customers and partners.

  • Work with internal applied researchers to improve their AI tools.

  • Advise design of new hardware generations.

What we need to see:

  • BS Computer Science, Computer Architecture, Electrical Engineering or a related field (or equivalent experience).

  • 6+ years of experience with C++ and Python programming.

  • Proven experience developing large software projects.

  • We require excellent verbal and written communication skills.

Ways to stand out from the crowd:

  • Experience in PyTorch.

  • Experience with parallel programming for CPU or GPU architectures.

  • We participate in the open source community.

  • Proven experience working with multidisciplinary teams.

NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most brilliant and talented people on the planet working for us. If you're creative, passionate, and self-motivated, we want to hear from you! NVIDIA is leading the way in groundbreaking developments in Artificial Intelligence, High-Performance Computing and Visualization. The GPU, our invention, serves as the visual cortex of modern computers and is at the heart of our products and services. Our work opens up new universes to explore, enables amazing creativity and discovery, and powers what were once science fiction inventions from artificial intelligence to autonomous cars. NVIDIA is looking for great people like you to help us accelerate the next wave of artificial intelligence.