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 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.
Job Description
Who you are
You have a strong background in compiler engineering and machine learning, with experience in developing and optimizing compilers for deep learning frameworks like TensorFlow, PyTorch, and JAX. Your expertise in instruction scheduling, memory allocation, and data transfer optimization allows you to tackle complex challenges in compiler technology. You are comfortable collaborating with cross-functional teams, including Runtime, Frameworks, and Hardware, to ensure system-wide performance optimization. You thrive in a fast-paced environment and are eager to contribute to cutting-edge innovations in machine learning.
Desirable
Experience with AWS services and custom hardware accelerators such as Inferentia and Trainium is a plus. Familiarity with graph partitioning and parallel programming will enhance your ability to contribute effectively to the team.
What you'll do
As a Machine Learning Compiler Engineer, you will be instrumental in designing and developing features for our deep learning compiler stack. You will work closely with a talented engineering team to optimize application models across various domains, including Large Language and Vision. Your responsibilities will include tackling crucial challenges in compiler technology, contributing to leading-edge design and research, and collaborating with cross-functional teams to ensure optimal performance of our machine learning systems. You will also engage in hardware-software co-design, working on new hardware bring-up and Instruction Set Architectures to enhance our compiler capabilities.
What we offer
At Amazon, you will be part of a dynamic team that is at the forefront of machine learning innovation. We provide a comprehensive benefits package, including equity and sign-on payments, along with a full range of medical and financial benefits. You will have the opportunity to work on groundbreaking projects that power solutions like Generative AI, making a significant impact in the field of machine learning. 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
Machine Learning Engineer
Amazon is hiring a Machine Learning Engineer II to work on AWS Machine Learning accelerators. You'll develop a deep learning compiler stack and optimize performance for complex neural net models. This position requires experience with AWS and popular ML frameworks.
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.
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 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.
Machine Learning Engineer
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.