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. This role focuses on developing and optimizing AWS's complete software stack for cloud-scale machine learning accelerators. You'll work with cutting-edge ML technologies, including large language models and vision transformers, while collaborating with chip architects and compiler engineers.

The position offers an opportunity to shape the future of machine learning infrastructure at AWS, working with technologies like PyTorch, TensorFlow, and distributed training libraries. You'll be responsible for performance tuning and optimization of ML models on AWS's custom silicon (Inferentia and Trainium).

AWS, as part of Amazon's Utility Computing organization, is at the forefront of cloud innovation, providing foundational services like S3 and EC2. The role combines deep technical expertise in software engineering with machine learning knowledge, making it ideal for someone passionate about both fields.

The company offers a strong emphasis on work-life harmony, with flexible working arrangements and a supportive culture. Amazon's commitment to diversity is evident through various employee-led affinity groups and ongoing learning experiences. Career growth is facilitated through mentorship programs and knowledge-sharing opportunities.

The compensation package is competitive, ranging from $151,300 to $261,500 based on location, plus equity and comprehensive benefits. This role represents an opportunity to work on cutting-edge ML infrastructure while being part of AWS's mission to be Earth's Best Employer.

Key technical aspects include working with distributed training frameworks, optimizing large language models, and developing solutions for AWS's custom ML accelerators. The ideal candidate will combine strong software development skills with deep ML knowledge to drive innovation in cloud-based machine learning infrastructure.