Annapurna Labs

Technology

SoftwareDevelopmentEngineer,NeuronCollectives

$165–224k Cupertino, California, United States FULL TIME
The Brief

“Software Development Engineer, Neuron Collectives at Annapurna Labs. Skills: Collective operations, AI training, Hardware/software integration. Enhance collective algorithms. Enhance topologies”

What You'll Achieve.

Optimize collective operations; Scale AI compute; Maximize performance; Utilize compute bandwidth; Utilize bus bandwidth

Industry & Context.

Technology
Problems you'll solve

Root cause analysis

What They're Looking For.

Must Have

Experience building complex software systems, Contribute to architecture and design, Bachelor's degree in computer science, Knowledge of engineering practices, Development experience in last 3 years, Embedded development in C/C++

Nice to Have

Master's degree in computer science, Hardware/software integration experience, Real-time systems experience, Collective communication algorithms familiarity, Distributed training frameworks familiarity

What You'll Do.

Enhance collective algorithms

Monitor processor metrics

Analyze processor metrics

Monitor firmware metrics

Analyze firmware metrics

Monitor workload metrics

Analyze workload metrics

Optimize collective operations

Co-optimize Trainium silicon

Develop C/C++ implementations

Optimize C/C++ implementations

Investigate improvements

Implement improvements

Build analysis frameworks

Maintain analysis frameworks

Build automation solutions

Maintain automation solutions

How You'll Work.

Team & Collaboration

Work closely hardware team

Free ATS check

Applying for this Software Development Engineer, Neuron Collectives role?

Most applicants get filtered before a human reads their resume. See if yours makes the cut.

ANONYMOUS · UNFILTERED

What do employees actually say about Annapurna Labs?

Real rants from real employees. Read before you apply.

Read Company Rants →