Software Engineer, Collective Communication

Onsite San Francisco

scalingfulltime

ashby

Match & cover letter

Create a profile to see your match and get a cover letter for this role.

Create profile and match

Job description

You will design and implement custom networking collectives for our training stack in San Francisco. This role involves collaborating with ML researchers and optimizing network performance for supercomputers.

Details

San Francisco, in-office 3 days a week
Relocation assistance offered
Experience with low level performance critical software

The work

Collaborate with ML researchers to implement collective operations in C++ and CUDA
Optimize training jobs for different network transports
Work on simulations for future supercomputer network designs