Sr. Software Engineer- AI/ML, AWS Neuron Apps

AWS Neuron is seeking a Senior Software Engineer to join their Machine Learning Applications (ML Apps) team, focusing on the complete software stack for AWS Inferentia and Trainium cloud-scale machine learning accelerators. This role sits at the intersection of cutting-edge ML infrastructure and large-scale distributed systems, working with massive language models like GPT-2 and GPT-3, stable diffusion, and Vision Transformers.

As part of AWS Utility Computing (UC), you'll contribute to foundational services like S3 and EC2, while working alongside chip architects and compiler engineers to optimize ML model performance. The position offers a unique opportunity to shape the future of machine learning infrastructure at AWS, working with state-of-the-art accelerator hardware and distributed training frameworks.

Amazon's culture emphasizes work-life harmony, inclusive team environments, and continuous learning. The company offers comprehensive benefits, mentorship opportunities, and career growth through various channels including employee-led affinity groups and ongoing learning experiences. The role combines deep technical expertise in ML systems with the scale and impact of AWS's cloud infrastructure.

The compensation package reflects Amazon's commitment to top talent, with a base salary range of $151,300 to $261,500 depending on location, plus equity and comprehensive benefits. This role requires strong software development skills combined with deep ML knowledge, making it ideal for engineers passionate about both high-performance computing and machine learning systems.