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

Annapurna Labs, now part of AWS, is seeking a Senior Machine Learning Engineer for their Distribute Training team working on AWS Neuron. This role focuses on developing and optimizing distributed training solutions for large-scale ML models including LLMs, Stable Diffusion, and Vision Transformers. The position involves working with cutting-edge technology like AWS Trainium and Inferentia accelerators, requiring expertise in distributed training libraries such as FSDP, Deepspeed, and Nemo. The team offers a collaborative environment with opportunities for mentorship and knowledge-sharing. AWS, as the world's leading cloud platform, provides comprehensive benefits, work-life harmony, and a strong commitment to diversity and inclusion. The role offers competitive compensation ranging from $151,300 to $261,500 based on location, plus additional benefits including equity and sign-on payments. This is an opportunity to work at the forefront of ML infrastructure, developing solutions that power the next generation of cloud computing.