Amazon’s Neuron Collectives team seeks a Software Developer to advance collective algorithms and topologies for AI training. You will optimize software-hardware co-design with Trainium silicon, verify bottlenecks using Neuron Explorer, and drive scalable collective operations across data centers.
You’ll collaborate with Annapurna Labs and infrastructure teams, building robust analysis frameworks and improving performance for large-scale ML workloads.
#J-18808-Ljbffr