Amazon Sr Software Development Manager AWS Neuron ML Frameworks 

United States, California, Cupertino 

508071548

As the Sr. Software Development Manager of ML Frameworks & Ecosystems you will lead the team that develops and extends Neuron support for leading ML frameworks including PyTorch and JAX. You will develop and deliver the framework plugins and libraries that enable a great user experience for developing and optimizing models on Trainium and Inferentia accelerators, and work closely with the open source ecosystem to drive improvements to enable models port seamlessly across accelerators.You will work closely with the Neuron compiler, training and inference optimization teams, ML model developers and users to deliver best performance on top AI models.You should have an established background in AI Frameworks and Machine Learning infrastructure such as PyTorch, PyTorch/XLA, and JAX. Experience with OpenXLA is a significant plus. You should have demonstrated ability to work with open source communities to influence future community direction, a strong technical understanding and a motivation to achieve results.Key job responsibilities
Responsible for the full life cycle of developing and releasing JAX and Pytorch framework support for AWS Neuron.Understand current and future directions of ML framework development, with a focus on enabling and optimizing the latest features of ML frameworks.A day in the life
You will work with the executive leadership and other senior management and technical leaders to define strategic directions and deliver new capabilities to ML model developers and users. You will work closely with your team to enhance our current framework support for the latest ML models and for top customers and to grow and advance your team's capabilities. You will solve challenges facing current users to enable the best performance on the latest accelerators. You will collaborate with the PyTorch and JAX community across the AI industry to drive ML framework technology forward.
Work/Life Balance
Mentorship & Career Growth

- 10+ years of engineering experience
- 5+ years of engineering team management experience
- 10+ years of planning, designing, developing and delivering consumer software experience
- Experience partnering with product or program management teams
- Experience managing multiple concurrent programs, projects and development teams in an Agile environment