Senior Deep Learning Software Engineer, PyTorch

NVIDIA is seeking a Senior Deep Learning Software Engineer, PyTorch to join their team. This role involves designing and building PyTorch components and tools that run efficiently on Super Computers with 1000s of GPUs. The successful candidate will collaborate with NVIDIA's hardware and software teams to improve network and GPU efficiency in PyTorch, design, build and support production AI solutions used by enterprise customers and partners, and work with internal applied researchers to improve AI tools.

Key responsibilities include:

• Designing and building PyTorch components for large-scale GPU systems
• Collaborating across teams to enhance PyTorch efficiency
• Developing production AI solutions
• Supporting internal research efforts

Requirements:

• MS in Computer Science or Engineering (or equivalent experience)
• 5+ years professional experience in High Performance Computing
• Proficiency in Python and C++ programming
• Experience with Thread and Distributed Parallel Programming
• Proven track record in developing large software projects
• Strong communication skills

Preferred qualifications:

• Familiarity with Machine Learning
• Experience with CUDA Programming and RDMA networking
• Python expertise
• Open-source community participation
• Experience working in multi-disciplinary teams

NVIDIA offers competitive salaries and a generous benefits package. The base salary range for this position is $180,000 - $339,250 USD, with additional equity and benefits. The company values diversity and is an equal opportunity employer.

Join NVIDIA to work on cutting-edge AI technology and contribute to the development of the world's best AI platform.