Senior ML Compiler Engineer – Neuron

Annapurna Labs (U.S.) Inc. in Seattle is seeking a senior software engineer for the Neuron compiler team.

You will design, implement, test, deploy and maintain software that enhances compiler performance, stability and user interfaces for ML workloads run on AWS Inferentia and Trainium accelerators. You will collaborate with chip architects and ML scientists to optimize models from PyTorch, TensorFlow and JAX for cost-efficient deployment, and you will engage with open-source communities to