Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training
Amazon · Seattle, WA · Software Development
About this role
Amazon is hiring a senior-level Software Engineer based in Seattle, WA. The posting calls out experience with AWS, TensorFlow, PyTorch, LLMs. Listed education preference: a bachelor's degree or equivalent.
- Role
- Software Engineer
- Function
- software engineering
- Level
- senior
- Track
- Individual contributor
- Employment
- Full-time
- Location
- Seattle, WA
- Education
- Bachelor's degree
- Department
- Software Development
- Posted
- Sep 30, 2025
More roles at Amazon
Job description
from Amazon careersAnnapurna Labs designs silicon and software that accelerates innovation. Customers choose us to create cloud solutions that solve challenges that were unimaginable a short time ago—even yesterday. Our custom chips, accelerators, and software stacks enable us to take on technical challenges that have never been seen before, and deliver results that help our customers change the world. AWS Neuron is the complete software stack for the AWS Trainium (Trn1/Trn2) and Inferentia (Inf1/Inf2) our cloud-scale Machine Learning accelerators. This role is for a Senior Machine Learning Engineer in the Distribute Training team for AWS Neuron, responsible for development, enablement and performance tuning of a wide variety of ML model families, including massive-scale Large Language Models (LLM) such as GPT-OSS, Quen and Llama, as well as Stable Diffusion, Vision Transformers (ViT) and many more. The ML Distributed Training team works side by side with chip architects, compiler engineers and runtime engineers to create, build and tune distributed training solutions with Trainium instances. Experience with training these large models using Pythorch is a must. Distributed training with awareness of strategies like FSDP (Fully-Sharded Data Parallel), PP, Context parallel. Distributed training libraries like torchtitan, torchtune , HF RL , DeepSeek etc are central to…