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-LjbffrAI Training Systems Engineer – Collective Compute in cupertino at Unknown Company
This position is listed as full time and onsite.