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

AWS Neuron is seeking a Senior Software Engineer to join their Machine Learning Applications team, focusing on the complete software stack for AWS Inferentia and Trainium cloud-scale machine learning accelerators. This role represents a unique opportunity to work at the intersection of machine learning and high-performance computing, developing solutions for massive-scale language models and advanced AI applications.

The position involves collaborating with chip architects and engineers to optimize distributed training solutions, working with cutting-edge ML models including GPT-2, GPT-3, stable diffusion, and Vision Transformers. You'll be responsible for implementing and enhancing distributed training and inference support in frameworks like PyTorch and TensorFlow, while ensuring maximum performance on AWS's custom silicon.

Amazon offers a comprehensive benefits package, including competitive salary ranging from $151,300 to $261,500 based on location, plus equity and sign-on bonuses. The company's inclusive culture is reflected in its 16 Leadership Principles and supported by various employee-led affinity groups. They emphasize work-life balance and provide extensive opportunities for professional growth and mentorship.

The ideal candidate will bring 5+ years of software development experience, strong ML knowledge, and expertise in distributed systems. You'll be joining a team that values knowledge sharing, mentorship, and professional development, working on technology that powers some of the most advanced AI applications in the cloud computing industry.