Software Engineer- AI/ML, Amazon Neuron Training

What you’d do

Implement and tune components of a distributed training stack for large-scale training, post-training, and reinforcement learning on Trainium instances, working with PyTorch and the Neuron software stack. Full-time role involving parallelism strategies, workload profiling, and building internal tooling, benchmarks, and tests. Found on the employer's careers page on Sep 29, 2026. Apply through the link.