Software Development Manager • Lead

Who you are

You have a strong background in software development and machine learning, with at least 5 years of experience leading engineering teams. Your technical expertise allows you to navigate complex challenges in developing machine learning products, and you have a proven track record of delivering customer-facing solutions. You are motivated by results and have a passion for innovation in the field of machine learning.

Your experience includes working with frameworks such as PyTorch and JAX, and you are familiar with distributed training libraries like FSDP and DDP. You understand the intricacies of machine learning architectures, including MoE architectures, and are eager to enable models that leverage cutting-edge technology. You thrive in collaborative environments and enjoy solving challenging technical problems that have not been addressed before.

Desirable

Experience with cloud-scale machine learning accelerators and a deep understanding of the full development life cycle for integrations and extensions in machine learning is a plus. Familiarity with AWS services and a customer-centric approach to product development will set you apart.

What you'll do

As the Software Development Manager, you will lead a talented team of engineers and managers in the Machine Learning Distributed Training team. Your responsibilities will include designing, implementing, testing, and maintaining innovative software solutions that enhance service performance and durability. You will ensure that the right products are built and delivered to customers, collaborating across diverse teams and projects to have a significant impact on AWS's global customer base.

You will be involved in the full development life cycle, guiding your team through the complexities of machine learning product development. This includes enabling models using advanced architectures and ensuring support for key ML functionalities in a combined chip/software platform. You will also be responsible for mentoring your team members, fostering a culture of innovation and excellence.

Your role will require you to solve challenging technical problems at every layer of the stack, from design to deployment. You will work closely with cross-functional teams to ensure that the solutions you develop meet customer needs and drive business results. Your leadership will be crucial in navigating the evolving landscape of machine learning technologies and ensuring that your team remains at the forefront of innovation.

What we offer

At Amazon, we offer a competitive compensation package that includes equity, sign-on payments, and a comprehensive range of medical and financial benefits. You will have the opportunity to work in a dynamic environment where your contributions will directly impact the future of machine learning at AWS. We encourage you to apply even if your experience doesn't match every requirement, as we value diverse perspectives and backgrounds in our teams.