Neuron Collectives Software Engineer, Trainium Collectives

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…