Senior Software Engineer - Automated Parallel Programming

Job Description

Join the PyTorch Team @ NVIDIA to design and build tools for AI practitioners deploying scalable AI applications to thousands of GPUs, working collaboratively with multi-disciplinary engineering teams.

Requirements

• MS or PhD in Computer Science, Computer Engineering, Electrical Engineering, or related field
• 4+ years of experience in parallel programming with NVIDIA CUDA or similar languages
• 4+ years of experience in C++ programming
• Experience in developing large software projects
• Excellent verbal and written communication skills

Responsibilities

• Crafting a code generation system to accelerate portions of a graph from a machine learning framework
• Optimizing GPU performance in PyTorch in collaboration with NVIDIA's hardware and software teams
• Designing, building, and supporting production AI solutions for enterprise customers and partners
• Collaborating with internal applied researchers to enhance AI tools
• Advising on the design of new hardware generations

Benefits

• Multiple relocation packages
• Two weeklong shutdowns (mid-summer and year-end) in the US (in addition to PTO)
• 8-week parental leave
• 9 Employee Resource Groups
• Annual bonus offering
• Flexible work arrangements
• Up to 6% 401K matching