Software Engineer, Collective Communication

Overview

Role involves designing and implementing custom networking collectives for training stacks.

Ideal candidate has experience in low-level performance-sensitive software and collective communication.

Locations

• United States, California, San Francisco

Requirements

• Experience with C++ and CUDA
• Background in low-level performance-critical software
• Familiarity with network simulation techniques

Responsibilities

• Collaborate with ML researchers
• Design and implement efficient collective operations
• Work on simulations for future network designs

Benefits

• Medical, dental, and vision insurance
• Mental health and wellness support
• 401(k) plan with matching
• Generous time off
• Paid parental leave
• Annual learning and development stipend