Software Engineer- AI/ML, AWS Neuron Distributed Training

AWS Neuron is seeking a Software Engineer to join their Machine Learning Applications team, focusing on distributed training solutions. This role at Annapurna Labs, an AWS company, involves working on the complete software stack for AWS Inferentia and Trainium cloud-scale machine learning accelerators. The position combines deep software development expertise with machine learning, requiring experience with frameworks like PyTorch/TensorFlow and distributed training libraries.

The role offers an opportunity to work on cutting-edge ML infrastructure, developing solutions for massive scale language models and various ML applications. You'll collaborate with cross-functional teams including chip architects and compiler engineers to optimize performance on AWS's custom silicon.

AWS provides a collaborative environment with strong emphasis on work-life balance and career growth. The team culture promotes diversity and inclusion, with various employee-led affinity groups and ongoing learning experiences. You'll have opportunities for mentorship and knowledge sharing within a team of varied experience levels.

The position offers competitive compensation based on geographic location, plus equity and comprehensive benefits. You'll be part of AWS's mission to revolutionize cloud infrastructure while working on technologies that impact millions of users worldwide. This role is perfect for someone passionate about both software engineering and machine learning, with a desire to work on large-scale distributed systems.