Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training

What you’d do

Optimize distributed AI/ML training performance on AWS Trainium, focusing on throughput, model FLOPs utilization, and time to convergence. Work across PyTorch, JAX, and the Neuron compiler and runtime to enable and tune large-scale training workloads. Full-time. Found on the employer's careers page on Sep 30, 2026. Apply through the link.