Sr. Software Development Manager, AWS Neuron Machine Learning Distributed Training - Cupertino, California, USA

AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machine
learning accelerators and the Trn1 and Inf1 servers that use them. As a Sr. SDM of Software Development for the Machine Learning Distributed Training team, you will be responsible for leading a strong team of engineers and managers to help design and deploy these new products. A successful candidate will have an established background in developing Machine Learning products with direct customer-facing experience, a strong technical ability and a motivation to achieve results. Experience in Machine Learning and software development is also a must.

Responsible for the full development life cycle of our integrations and extensions for inference and training support in Pytorch, XLA, JAX as well as distributed training libraries like FSDP, DDP and others.
Characterization, enablement and development of existing and future massive-scale ML models like GPT3 as well as BERT, ViT, LLama, Stable Diffusion and more
Lead the way to ensure support for key ML functionality in a combined chip / software platform
Ensure the right thing is being built and delivered to customers



Key job responsibilities
Responsible for the full development life cycle of our integrations and extensions for training support in Pytorch, XLA, Tensorflow as well as distributed training libraries like FSDP, DDP and others.
Characterization, enablement and development of existing and future massive-scale ML models like GPT3 as well as BERT, ViT, Stable Diffusion and more
Lead the way to ensure support for key ML functionality in a combined chip / software platform
Ensure the right thing is being built and delivered to customers

A day in the life
You will work with the executive leadership and other senior …