Software Engineer- AI/ML, Amazon Neuron Training

What you’d do

Optimize distributed training performance on Trainium hardware, tuning large-scale workloads across the PyTorch and Neuron software stacks. Work spans parallelism strategies, reduced-precision formats, and end-to-end profiling with compiler and runtime teams. Full-time in Seattle. Found on the employer's careers page on Sep 30, 2026. Apply through the link.