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 cloud-scale machine learning accelerators, specifically working with AWS Inferentia and Trainium technologies. The position involves working with cutting-edge ML models including GPT-2, GPT-3, stable diffusion, and Vision Transformers.

The role combines deep technical expertise in both software engineering and machine learning, requiring hands-on experience with distributed training using frameworks like PyTorch and TensorFlow. You'll collaborate closely with chip architects and compiler engineers to build and optimize distributed training solutions for AWS's specialized ML hardware.

Amazon offers a comprehensive benefits package and emphasizes work-life balance. The team culture promotes knowledge sharing, mentorship, and professional growth. With Amazon's 16 Leadership Principles as a foundation, the environment encourages diverse perspectives and continuous learning.

The position offers competitive compensation ranging from $151,300 to $261,500 based on location and experience, plus additional benefits including equity and sign-on payments. This is an excellent opportunity for experienced engineers passionate about machine learning infrastructure and looking to make an impact at scale.