Senior Software Engineer, PyTorch - Deep Learning

at Nvidia
USD 148,000-287,500 per year
SENIOR
✅ Hybrid

SCRAPED

Used Tools & Technologies

Not specified

Required Skills & Competences ?

Python @ 4 Machine Learning @ 3 Hiring @ 4 Communication @ 7 Parallel Programming @ 4 PyTorch @ 4 CUDA @ 4 GPU @ 4

Details

NVIDIA is hiring a Senior Deep Learning Software Engineer specialized in PyTorch to design and build scalable AI applications across thousands of GPUs. This role involves collaboration with NVIDIA’s hardware and software teams and the international PyTorch community to develop a leading AI platform.

Responsibilities

  • Design and build PyTorch components optimized for supercomputers with thousands of GPUs.
  • Collaborate with hardware and software teams at NVIDIA to enhance GPU performance in PyTorch.
  • Develop and maintain production AI solutions for enterprise customers and partners.
  • Work together with internal applied researchers to improve AI tools.

Requirements

  • MS in Computer Science or Engineering, or equivalent experience.
  • Over 3 years of professional experience in High Performance Computing.
  • Proficient in C++ programming.
  • Experience with Thread and Distributed Parallel Programming.
  • Proven track record in developing large software projects.
  • Strong verbal and written communication skills.

Ways to Stand Out

  • Familiarity with Machine Learning concepts.
  • Background in CUDA programming.
  • Experience with Python.
  • Participation in the open source community.
  • Experience working with multi-disciplinary teams.

Benefits

  • Competitive salary ranging from 148,000 to 287,500 USD annually, depending on location, experience, and comparable roles.
  • Eligibility for equity and comprehensive benefits.
  • Work in a highly innovative environment focused on Deep Learning, AI, and Autonomous Vehicles.

Additional Information

  • Job type: Full time
  • Location: Santa Clara, California, United States
  • NVIDIA emphasizes diversity and equal opportunity employment.

#deeplearning