Senior AI/ML Distributed Training Engineer - RSUs

Amazon Inc. is seeking a Sr. Software Engineer- AI/ML for AWS Neuron Distributed Training in Cupertino. You will optimize large‑scale distributed training across the Neuron stack, spanning PyTorch, JAX, and the compiler/runtime, while shaping parallelism, numerics, and performance across a broad hardware/software boundary. You will mentor engineers, collaborate with customers, and participate in open source collaborations to ensure scalable, efficient model training on Trainium. #J-18808-Ljbffr Amazon Inc.