Software Engineer II, Annapurna Labs

Annapurna Labs, acquired by AWS in 2015, serves as the infrastructure provider for AWS, focusing on silicon engineering, hardware design and verification, software, and operations. This role is for a Software Engineer II position within the AWS Neuron team, which develops the complete software stack for AWS Inferentia and Trainium cloud-scale machine learning accelerators.

The position offers an exciting opportunity to lead the development of machine learning tools for running, optimizing, and analyzing ML workloads. The role combines technical leadership with hands-on development, requiring expertise in machine learning tools, profiling, optimization, and various instruction set architectures.

Working in a startup-like environment within AWS, you'll be part of a team that has delivered impressive products including AWS Nitro, ENA, EFA, Graviton, and F1 EC2 Instances. The team culture emphasizes knowledge-sharing, mentorship, and inclusive practices, with a strong focus on work-life harmony.

AWS values diverse experiences and provides comprehensive benefits, including medical coverage, financial benefits, and career development opportunities. The hybrid work model offers flexibility while maintaining opportunities for meaningful in-person collaboration.

This role is ideal for someone who wants to work at the intersection of machine learning and infrastructure, contributing to tools that will shape the future of cloud computing while being part of a supportive, innovative team that values both technical excellence and personal growth.