Neuron Collectives Software Engineer, Trainium Collectives

Annapurna Labs Inc. · Monte Vista, CA

Verified today

In the employer's feed today

“Verified” means one of three checks: the employer’s own careers page, their job feed, or the employer’s own word. The line above says which one, and when.

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…

The feed this listing came from sends a shortened description. See the full posting on the employer’s site.

Listed under Technology & IT jobs.

Similar openings

Browse all jobs →