Software Engineer- AI/ML, AWS Neuron

AWS Neuron is seeking a talented Software Engineer to join their Machine Learning Applications (ML Apps) team. This role focuses on developing and optimizing the complete software stack for AWS Inferentia and Trainium cloud-scale machine learning accelerators. You'll work on cutting-edge ML technologies, including large language models like LLama3 and Mixtral, while collaborating with chip architects and compiler engineers.

The position offers an exciting opportunity to shape the future of machine learning infrastructure at AWS, working with distributed training systems and performance optimization. You'll be part of a team that values work-life balance, mentorship, and professional growth. The role combines deep technical expertise in both software development and machine learning, making it perfect for engineers passionate about AI infrastructure.

Amazon's inclusive culture, guided by their 16 Leadership Principles, provides a supportive environment with employee-led affinity groups and ongoing learning opportunities. The compensation package is competitive, ranging from $129,300 to $223,600 based on location, plus comprehensive benefits and potential equity compensation.

This role is ideal for someone with strong software development skills and ML knowledge who wants to work on large-scale, impactful projects in the AI/ML infrastructure space. You'll be at the forefront of developing solutions that power some of the most advanced ML applications in cloud computing.