Senior Software Development Engineer, Neuron Collectives, Annapurna Labs

Job overview

The Senior Software Development Engineer, Neuron Collectives, Annapurna Labs role is an onsite position based in Cupertino, CA. Compensation is not specified in the posting, and this role supports the AWS Neuron Collectives team in optimizing AI compute performance for Trainium hardware. This role exists to enhance collective communication algorithms and topologies for modern LLM training. The engineer will work at the intersection of hardware, firmware, and distributed systems to maximize compute and bus bandwidth utilization across data centers.

What you'll do

• Enhance collective algorithms
• use Neuron Explorer to identify bottlenecks
• monitor processor and firmware metrics
• optimize collective operations
• co-optimize software and Trainium silicon
• develop C/C++ implementations of communication patterns.

What we're looking for

Skills & competencies

full-time

day shift

cupertino ca

ai training

machine learning

distributed systems

c++

trainium

annapurna labs

onsite

Work arrangement

• Weekend coverage required

Benefits & perks

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

Why this role

Focuses on high-performance AI training hardware and collective communication primitives.

About the employer

Annapurna Labs (U.S.) Inc. is hiring for this role. Industry: Custom Computer Programming Services. Sector: 54.