Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training

AWS Utility Computing (UC) is at the forefront of cloud innovation, developing cutting-edge solutions through Annapurna Labs. This senior role focuses on the AWS Neuron software stack, working with AWS Inferentia and Trainium cloud-scale machine learning accelerators. The position combines advanced software engineering with machine learning expertise, requiring deep knowledge of distributed training systems and major ML frameworks.

The role involves collaborating with a diverse team of chip architects and engineers to optimize performance for large language models and AI systems. You'll be working on groundbreaking technology that powers AWS's machine learning infrastructure, including work with models like GPT-2, GPT-3, and stable diffusion.

AWS offers a comprehensive benefits package, including competitive salary, equity compensation, and full medical coverage. The team culture emphasizes mentorship, knowledge-sharing, and work-life harmony. You'll be part of an inclusive environment that values diverse experiences and perspectives.

The position provides unique opportunities to work with cutting-edge AI/ML technology while contributing to systems that help customers solve previously impossible challenges. You'll have access to continuous learning opportunities, career development resources, and the chance to work with some of the most advanced cloud computing and AI technologies in the industry.

This role is ideal for experienced engineers who are passionate about machine learning, distributed systems, and high-performance computing. You'll be part of AWS's mission to innovate in the cloud computing space while enjoying the benefits of working for a global technology leader.