Sr. Machine Learning Compiler Engineer, Annapurna Labs
London, United Kingdom
Amazon
chip for ML training - delivers industry-leading performance and cost efficiency at scale. These are powered by the AWS Neuron Software Development Kit (SDK), which includes a deep learning compiler, runtime, and native integration with popular ML frameworks like PyTorch, TensorFlow, and MXNet. Customers such as Snap, Autodesk, Amazon Alexa, and Amazon Rekognition are already benefiting from this technology … the founding team, shaping our culture and technical direction locally while contributing to a global mission. The Tel Aviv team will play a key role in developing the Neuron compiler stack - translating high-level neural network models into high-performance execution on AWS custom hardware. Key job responsibilities The Team: As part of Amazon Annapurna Labs, the AWS Neuron … performance and scalability of ML workloads across the cloud. Our focus: creating a robust, high-performance toolchain that accelerates innovation in Generative AI. You: As a Senior Machine Learning Compiler Engineer in Tel Aviv, you'll lead the development of cutting-edge compiler technologies, optimize large-scale ML workloads, and bring new hardware to life. You'll More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted: