
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 hiring a Senior Machine Learning Engineer to work on AWS Neuron, focusing on optimizing ML performance on custom-built hardware. You'll utilize AWS tools and frameworks like TensorFlow and PyTorch. This role requires significant experience in machine learning and compiler development.
Job Description
Who you are
You have 5+ years of experience in machine learning and compiler development, with a strong background in optimizing neural network models for performance. Your expertise includes working with frameworks such as TensorFlow, PyTorch, and MXNet, and you are familiar with the intricacies of deep learning compilers. You thrive in collaborative environments and enjoy tackling complex challenges that push the boundaries of technology.
You possess a deep understanding of AWS services and how they can be leveraged to enhance machine learning capabilities. Your technical skills are complemented by your ability to communicate effectively with cross-functional teams, ensuring that project goals are met efficiently. You are passionate about innovation and are eager to contribute to cutting-edge projects that have a significant impact on the industry.
What you'll do
As a Senior Machine Learning Engineer at Amazon, you will be responsible for developing and optimizing the deep learning compiler stack for AWS Neuron. Your work will involve translating neural network descriptions into executable code, ensuring that our custom-built hardware delivers the best performance possible. You will collaborate closely with silicon engineers and software developers to enhance the AWS Machine Learning accelerators, contributing to the overall success of AWS's generative AI tools.
You will engage in performance tuning and optimization of complex neural net models, working with various machine learning frameworks to ensure seamless integration with AWS services. Your role will also include mentoring junior engineers and sharing your knowledge of best practices in machine learning and compiler design. You will have the opportunity to work on projects that are at the forefront of AWS innovation, impacting customers across various segments.
What we offer
Amazon provides a competitive salary range of $193,300.00 - $261,500.00 USD annually, along with comprehensive benefits including health insurance, retirement plans, and paid time off. You will be part of a dynamic team that values innovation and collaboration, with opportunities for professional growth and development. Join us in shaping the future of machine learning at AWS.
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

Machine Learning Engineer
Amazon is hiring a Senior Machine Learning Engineer to work on AWS Neuron, developing a deep learning compiler stack for optimizing neural network models. You'll utilize AWS tools and frameworks like TensorFlow and PyTorch. This position requires significant experience in machine learning and compiler development.

Machine Learning Engineer
Amazon is hiring a Machine Learning Engineer for the AWS Neuron team to build next-generation compilers for deep learning models. You'll work with technologies like TensorFlow and PyTorch to optimize performance on custom chips. This position requires experience in compiler optimization and machine learning frameworks.

Machine Learning Engineer
Amazon is hiring a Machine Learning - Compiler Engineer II to develop the Neuron compiler for optimizing ML models on AWS Inferentia and Trainium. You'll work with technologies like Python, TensorFlow, and PyTorch, and the role requires experience in compiler optimization.

Ml Kernel Performance Engineer
Amazon is hiring an ML Kernel Performance Engineer to optimize performance for AWS's custom ML accelerators. You'll work with technologies like AWS, Python, TensorFlow, and PyTorch. This position requires expertise in machine learning and high-performance computing.

Machine Learning Engineer
Amazon is hiring a Machine Learning Compiler Engineer to develop a state-of-the-art deep learning compiler stack. You'll work with frameworks like TensorFlow and PyTorch, optimizing models for AWS's custom Machine Learning accelerators. This position requires experience in compiler technology and deep learning systems.