
About Amazon
The everything store and cloud computing leader
Key Highlights
- Headquartered in South Lake Union, Seattle, WA
- Over 1.5 million employees worldwide
- Leading cloud services through Amazon Web Services (AWS)
- Acquired Whole Foods, Twitch, and Ring
Amazon, headquartered in South Lake Union, Seattle, WA, is the world's largest online retailer and a leader in cloud computing through Amazon Web Services (AWS). With over 1.5 million employees globally, Amazon operates in various sectors, including AI with its Alexa devices and a vast marketplace k...
🎁 Benefits
Amazon offers competitive salaries, stock options, generous PTO policies, and comprehensive health benefits. Employees also have access to a learning ...
🌟 Culture
Amazon's culture is driven by customer obsession and a focus on innovation. The company encourages employees to think big and move fast, fostering an ...
Skills & Technologies
Overview
Amazon is seeking a Software Engineering Manager for their ML Accelerators team at AWS Neuron. You'll lead a team of compiler engineers to design and implement machine learning compiler solutions using technologies like Python, TensorFlow, and PyTorch. This role requires strong leadership and technical expertise in machine learning.
Job Description
Who you are
You have 5+ years of experience in software engineering, with a strong focus on machine learning and compiler design. Your background includes leading teams and driving technical projects from conception to implementation, ensuring high-quality outcomes. You possess deep knowledge of machine learning frameworks such as TensorFlow, PyTorch, and JAX, and you are comfortable working with complex systems and architectures. Your experience with AWS services and cloud technologies enables you to leverage the full potential of cloud-based solutions in your projects.
You are a natural leader who thrives in a collaborative environment. You enjoy mentoring and developing talent within your team, fostering a culture of continuous learning and innovation. You understand the importance of balancing technical excellence with team dynamics, and you are skilled at navigating cross-functional collaborations to achieve project goals. You are passionate about driving advancements in machine learning technology and are eager to contribute to the development of next-generation solutions.
Desirable
Experience with hardware-software co-design and optimization techniques is a plus. Familiarity with silicon design processes and pre-silicon validation will enhance your ability to influence design decisions effectively. You are also open to exploring new technologies and methodologies that can improve the performance and efficiency of machine learning applications.
What you'll do
In this role, you will lead a talented team of compiler engineers focused on developing advanced machine learning compiler solutions. You will drive the design and implementation of optimization techniques that enhance the performance of machine learning models on AWS infrastructure. Your responsibilities will include overseeing the hardware bring-up for next-generation chips, ensuring that the software stack integrates seamlessly with the underlying hardware.
You will collaborate closely with cross-functional teams, including silicon design and hardware verification, to influence pre-silicon design decisions. Your leadership will be crucial in guiding your team through complex technical challenges, ensuring that they have the resources and support needed to succeed. You will also be responsible for setting clear objectives and performance metrics for your team, fostering a culture of accountability and excellence.
As part of your role, you will engage with internal and external customers to understand their needs and gather feedback on the compiler solutions being developed. This customer-centric approach will help you prioritize features and enhancements that deliver maximum value. You will also stay abreast of industry trends and advancements in machine learning and compiler technologies, ensuring that your team remains at the forefront of innovation.
What we offer
Amazon offers a competitive compensation package, including equity and sign-on bonuses, along with a comprehensive range of medical, financial, and other benefits. You will have the opportunity to work in a dynamic and enriching environment that encourages experimentation and innovation. Join a multidisciplinary team that is driving significant advancements in machine learning technology and making a real impact on the future of cloud computing. We encourage you to apply even if your experience doesn't match every requirement, as we value diverse perspectives and backgrounds.
Interested in this role?
Apply now or save it for later. Get alerts for similar jobs at Amazon.
Similar Jobs You Might Like
Based on your interests and this role

Software Development Manager
Amazon is hiring a Software Development Manager for the Neuron Tools team to lead the development of high-performance monitoring and profiling tools for machine learning applications. You'll work with C++, PyTorch, and TensorFlow to optimize AI workloads. This position requires a strong background in AI/ML tools and performance analysis.

Machine Learning Engineer
Amazon is hiring a Senior Machine Learning Compiler Engineer for AWS Neuron to optimize ML performance on custom-built hardware. You'll work with TensorFlow, PyTorch, and MXNet to develop a deep learning compiler stack. This position requires expertise in machine learning and compiler technologies.

Software Development Manager
Amazon is hiring a Software Development Manager for the AWS Neuron team to lead the development of machine learning compilers and software stacks. You'll work with technologies like AWS, PyTorch, and TensorFlow in Toronto.

Machine Learning Engineer
Amazon is hiring a Machine Learning Engineer for the AWS Neuron team to develop and optimize building blocks for machine learning applications. You'll work with technologies like Python, PyTorch, and AWS to enhance inference performance. This role requires experience in optimizing LLM inference performance.

Software Development Manager
Amazon is hiring a Software Development Manager for AI Inference Technology to lead a team optimizing large language models for Trainium. You'll work with AWS, PyTorch, and distributed systems. This position requires a strong background in machine learning and team management.