Software Development Manager, AWS Neuron Frameworks

Job description

We're seeking a Software Development Manager to lead our Frameworks team within AWS Neuron, the software stack powering AWS Inferentia and Trainium machine learning accelerators. This role combines technical leadership, team management, and strategic open-source collaboration to shape the future of machine learning acceleration at AWS. As the Software Development Manager, you will lead and mentor a high-performing team of software engineers while driving the development and maintenance of critical Neuron framework components. You'll drive cross-functional collaboration with compiler, runtime, and kernel development teams to ensure seamless integration of Neuron with major machine learning frameworks. You will also contribute technically by reviewing designs and implementing features. A crucial aspect of your role will be building and nurturing strategic relationships with open-source communities, particularly with JAX, OpenXLA, and PyTorch/XLA. You'll work closely with these communities to align framework development roadmaps with Neuron's strategic objectives, ensuring our customers have access to the latest ML framework innovations. In this role, you'll be instrumental in shaping how developers interact with AWS's machine learning accelerators, ensuring our customers can easily adopt and benefit from the latest advancements in ML frameworks. The position offers an opportunity to work at the intersection of cloud computing, machine learning,…