The Automated Reasoning Group in the AWS Neuron Compiler team seeks an Applied Scientist to work at the intersection of AI and program analysis. This role focuses on enhancing code quality in their state-of-the-art deep learning compiler stack, which optimizes application models across various domains, including Large Language and Vision, from frameworks like PyTorch, TensorFlow, and JAX.
The position involves working with AWS's custom-built Machine Learning accelerators, Inferentia and Trainium, representing the forefront of AWS innovation for advanced ML capabilities and Generative AI. As an Applied Scientist, you'll be crucial in designing and implementing analyzers for ML compiler stages, publishing research, and mentoring talented team members.
AWS offers a comprehensive environment for growth and innovation, being the world's leading cloud platform. The role sits within the AWS Utility Computing (UC) organization, which provides foundational services like S3 and EC2. The company culture strongly emphasizes learning, curiosity, and inclusion, with various employee-led affinity groups and ongoing learning experiences.
The position offers competitive compensation ranging from $136,000 to $223,400 per year, depending on location, plus additional benefits and equity opportunities. AWS values diverse experiences and provides extensive career development resources, mentorship, and a strong focus on work-life harmony. This is an excellent opportunity for someone passionate about compiler technology, machine learning, and advancing the state of the art in AI systems.