Senior ML Compiler Engineer - NKI for ML Accelerators

Amazon Annapurna Labs seeks a Sr. Machine Learning Compiler Engineer to lead the development and scaling of a compiler for the world’s largest ML workloads on AWS Trainium.

You will architect critical features, publish cutting-edge work, and mentor a team of engineers, partnering closely with AWS ML teams. The role favors ML accelerators and a background in machine learning; relocation to Seattle, Cupertino, or Austin may be required.