Neuron Collectives Software Engineer, Trainium Collectives

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…