Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs
Summary
Develops and optimizes compiler tooling for AWS Neuron to accelerate ML inference and training on AWS Inferentia and Trainium chips.
Description The Product: AWS Machine Learning accelerators are at the forefront of AWS innovation and one of several AWS tools used for building Generative AI on AWS. The Inferentia chip delivers best-in-class ML inference performance at the lowest cost in cloud. Trainium will deliver the best-in-class ML training performance with the most teraflops (TFLOPS) of compute power for ML in the cloud. This is all enabled by cutting edge software stack, the AWS Neuron Software Development Kit (SDK), w…