Software Development Manager - ML Performance Tooling and Benchmarking, AWS Neuron, Annapurna Labs

The AWS Machine Learning accelerators, including Inferentia and Trainium, deliver best‑in‑class inference and training performance. They are backed by the AWS Neuron Software Development Kit, which optimizes neural network models for these chips.

The Amazon Annapurna Labs team focuses on innovation in silicon and software. It encompasses silicon engineering, hardware design and verification, software and operations, and places emphasis on small, agile, cross‑disciplinary teams.

Responsibilities

• Lead a team of compiler engineers to develop, deploy, and scale a compiler targeting AWS Inferentia and Trainium.
• Act as a technical partner with AWS ML services teams, contributing to pre‑silicon design and bringing new products, optimizations, and features to market.
• Leverage deep knowledge of resource management, scheduling, cod...