Neuron Collectives Software Engineer, Trainium Collectives

Job overview

The Neuron Collectives Software Engineer role is an onsite position based in Cupertino, CA. Compensation is not specified in the posting, and this role supports the Annapurna Labs team in building networking solutions for ML and HPC workloads. This role exists to enhance collective algorithms and topologies for optimal training performance. The engineer will optimize collective operations to scale AI compute across the data center.

What you'll do

• Enhance collective algorithms and topologies for optimal training performance.
• Use tools to identify bottlenecks in compute and bus bandwidth.
• Monitor and analyze processor, DMA, and firmware metrics.
• Optimize collective operations to scale AI compute.
• Develop and optimize C/C++ implementations of collective communication patterns.

What we're looking for

Skills & competencies

full-time

cupertino ca

software engineering

machine learning

networking

hpc

collectives

c++

optimization

onsite

no on-call

no travel

Work arrangement

• Weekend coverage required

Benefits & perks

• Health insurance, 401(k) matching, paid time off, parental leave, sign-on payments, and RSUs.

Why this role

Focus on collective communication primitives for large-scale AI training.

About the employer

Annapurna Labs (U.S.) Inc. is hiring for this role. Industry: Computing Infrastructure Providers, Data Processing, Web Hosting, and Related Services. Sector: 51.