Lead Software Engineer - AI/ML, AWS Neuron Distributed Training

At Annapurna Labs, we design silicon and software that drives innovation forward. Our customers rely on us to develop cloud solutions that tackle challenges previously deemed insurmountable. Our custom chips and software enable us to confront unprecedented technical hurdles and deliver transformative results. The AWS Neuron software stack supports our cloud-scale Machine Learning accelerators, Trainium and Inferentia. As a member of the Distributed Training team, you will collaborate closely with chip architects, compiler engineers, and runtime engineers, focusing on the development, enablement, and fine-tuning of various ML model families. Originally a startup acquired by AWS in 2015, we now integrate seamlessly into AWS's infrastructure offerings. Our team comprises diverse skills across silicon engineering, hardware design, software, and operations. We are committed to fostering an environment of mentorship and knowledge-sharing among our team members. At AWS, we value diverse backgrounds and experiences, welcoming applicants from all paths, even if they do not meet every qualification. Our inclusion and flexibility principles ensure a supportive cultural environment where every individual can thrive. As we strive to be Earth’s Best Employer, expect abundant growth opportunities through mentorship and shared knowledge. We offer a comprehensive compensation package that reflects the labor market, integrating salary and benefits tailored to individual candidates.