Senior Software Developer - AI/ML Distributed Systems

Annapurna Labs designs innovative silicon and software solutions. Our clients come to us for cloud solutions that tackle previously unimaginable challenges. Our specialized chips, accelerators, and software stacks empower us to confront technical obstacles that have never been encountered before, delivering results that assist our clients in making significant changes in the world. As a part of the AWS Neuron team, I will play a crucial role in the Distributed Training team for AWS Neuron, focusing on the development, enablement, and performance enhancement of various machine learning model families. This includes large-scale language models like GPT-OSS, Quen, and Llama, as well as models such as Stable Diffusion and Vision Transformers. Our team at Annapurna Labs, which became part of AWS in 2015, encompasses various disciplines, including silicon engineering, hardware design, software, and operations. We strive to create an environment that encourages knowledge-sharing and mentorship, emphasizing career growth and project assignments that help team members develop their engineering skills. At AWS, we embrace diverse experiences and welcome candidates from unconventional backgrounds to apply, even if they do not meet every qualification outlined. We value a culture of inclusion and support work-life balance, believing that employee well-being fosters better outcomes in our work. We also provide ongoing mentorship and resources to help team members become more well-rounded professionals. Overall, we are committed to fostering a supportive and inclusive workplace that empowers every employee to contribute to our customers' success.