Posted in

Senior Software Engineer – Pytorch – Deep Learning

Senior Software Engineer – Pytorch – Deep Learning

CompanyNVIDIA
LocationRemote, OR, USA, Santa Clara, CA, USA, New York, NY, USA
Salary$148000 – $287500
TypeFull-Time
DegreesMaster’s
Experience LevelSenior

Requirements

  • MS in Computer Science or Engineering (or equivalent experience)
  • 3+ years professional experience in High Performance Computing
  • Proficient with C++ programming
  • Demonstrated experience with Thread and Distributed Parallel Programming
  • Demonstrated experience developing large software projects
  • Strong verbal and written communication skills

Responsibilities

  • Design and build PyTorch components that run efficiently on Super Computers with 1000s of GPUs
  • Collaborate with NVIDIA’s hardware and software teams to improve the overall 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

Preferred Qualifications

  • Familiarity with Machine Learning
  • Background with CUDA Programming
  • Experience with Python
  • Participation in the open source community
  • Demonstrated experience working with multi-disciplinary teams