IT Jobs · United States
Neuron Collectives Software Engineer, Trainium Collectives
Annapurna Labs Inc. · Monte Vista, Santa Clara County, United States
177,218 – 177,218/ year
Job Description As a Neuron Collectives Software Developer, you will:\n \n \n* Enhance collective algorithms and topologies for optimal training performance\n \n* Use tools like Neuron Explorer to identify bottlenecks in compute and bus bandwidth utilization\n \n* Monitor and analyze processor, DMA, firmware, and workload metrics\n \n* Optimize collective operations to scale AI compute across the data center through low level device driver development\n \n* Work closely with the hardware team t…
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 →