Neuron Collectives Software Engineer, Trainium Collectives
Neuron Collectives Software Engineer, Trainium Collectives is an it role at Annapurna Labs Inc. in Monte Vista, Santa Clara County.
Annapurna Labs Inc.Monte Vista, Santa Clara County
$176,205–$176,205 a year
- Category
- IT Jobs
- Posted
- Sep 10, 2026
- Location
- Monte Vista, Santa Clara County
About the role
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…
This is the employer's posting preview; the full text is on their site.
About the employer
Annapurna Labs Inc. — view employer profile →Hiring in Monte Vista, Santa Clara County
Look up this employerRelated roles in IT Jobs
Skills this role needs (estimated)
Train these on your Skills page — each one has drills with XP.
Listing via Adzuna — applying takes you to the original posting.