Software Development Engineer, Annapurna Labs, AWS Neuron Frameworks

AWS Neuron is seeking a Software Development Engineer to join their Frameworks team, focusing on the complete software stack for AWS Inferentia and Trainium cloud-scale machine learning accelerators. This role combines cutting-edge ML infrastructure development with open-source community engagement, making it perfect for engineers passionate about machine learning frameworks and systems.

The position offers an opportunity to work with PyTorch and JAX frameworks, optimizing them for AWS's accelerator architectures. You'll be part of AWS's Utility Computing organization, which powers foundational services like S3 and EC2. The role requires strong technical skills in C++ and Python, along with deep ML knowledge.

Amazon offers a comprehensive benefits package and emphasizes work-life harmony. The company culture strongly supports diversity, inclusion, and continuous learning through employee-led affinity groups and ongoing educational experiences. Career growth is facilitated through mentorship programs and knowledge-sharing initiatives.

The compensation is competitive, ranging from $129,300 to $223,600 based on location, plus additional benefits including equity and sign-on payments. You'll be working with a collaborative team that values knowledge-sharing and mentorship, making it an ideal environment for professional growth in the ML infrastructure space.

The role is based in Cupertino, CA, and offers the opportunity to work on technology that powers AWS's machine learning services used by customers ranging from startups to Global 500 companies. If you're passionate about ML frameworks, system optimization, and want to impact how machine learning models are developed and deployed at scale, this position offers an excellent opportunity to make a significant impact.