ML Compiler Engineer II - Neuron Kernel Interface , Annapurna Labs

Market insights

Stack

Apply

Upload your CV — we will connect you with the employer directly through our pool.

Description

The AWS Neuron Compiler team is actively seeking skilled compiler engineers to join our efforts in developing a state-of-the-art deep learning compiler stack. This stack is designed to optimize application models across diverse domains, including Large Language and Vision, originating from leading frameworks such as PyTorch, TensorFlow, and JAX. Your role will involve working closely with our custom-built Machine Learning accelerators, including Inferentia/Trainium, which represent the forefron…

Employer contacts (email/phone/telegram) are hidden from the public preview — send your CV, and we will connect you directly.