IT Jobs · United States
Neuron Collectives Software Engineer, Trainium Collectives
Annapurna Labs Inc. · Monte Vista, Santa Clara County, United States
176,205 – 176,205/ year
As a Neuron Collectives Software Developer, you will: * Enhance collective algorithms and topologies for optimal training performance * Use tools like Neuron Explorer to identify bottlenecks in compute and bus bandwidth utilization * Monitor and analyze processor, DMA, firmware, and workload metrics * Optimize collective operations to scale AI compute across the data center through low level device driver development * Work closely with the hardware team to co-optimize software and Trainium sil…
Before you apply
Stand out at Annapurna Labs Inc.
Don't send a generic resume. Generate your Known Guy Trust Badge and attach it to your CV or cover letter — then apply.
Already have your badge? Apply on company website →