Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training

AWS Utility Computing (UC) is at the forefront of cloud innovation, developing cutting-edge solutions in the AWS ecosystem. This senior role within the Machine Learning Applications (ML Apps) team for AWS Neuron focuses on developing and optimizing machine learning solutions using AWS's custom silicon accelerators. The position involves working with state-of-the-art ML models, including GPT-series and Vision Transformers, while collaborating with chip architects and compiler engineers.

The role demands expertise in distributed training frameworks like FSDP and Deepspeed, combined with strong software development skills. You'll be working on AWS Neuron, the complete software stack for AWS Inferentia and Trainium cloud-scale machine learning accelerators. This position offers exposure to Amazon's growing suite of generative AI services and cutting-edge cloud computing technologies.

AWS values diverse experiences and maintains an inclusive culture through employee-led affinity groups and ongoing learning experiences. The team emphasizes knowledge-sharing, mentorship, and career growth, with senior members providing one-on-one mentoring and thorough code reviews. Work-life harmony is prioritized, ensuring success at work doesn't come at the expense of personal life.

The compensation package is comprehensive, including competitive base pay, equity, sign-on payments, and various benefits. The role offers an opportunity to work with advanced ML technologies while contributing to AWS's mission of being Earth's Best Employer. Join a team that's pushing the boundaries of what's possible in cloud computing and machine learning acceleration.