Software Engineer- AI/ML, AWS Neuron Distributed Training - Performance Optimization

What you’d do

Optimize distributed training performance on AWS Trainium hardware, focusing on training throughput and efficiency across the Neuron software stack. Work with PyTorch, JAX, and the Neuron compiler and runtime to tune large-scale training workloads. Full-time. Found on the employer's careers page on Sep 30, 2026. Apply through the link.