Sr. Software Engineer- AI/ML, Amazon Neuron Training

What you’d do

Lead work adding distributed training and post-training support to PyTorch and JAX for Trainium accelerators, tuning large-scale training and reinforcement learning workloads. Full-time role profiling performance and contributing upstream to open source frameworks. Found on the employer's careers page on Sep 29, 2026. Apply through the link.