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 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.
Job Description
Who you are
You have a strong background in machine learning and compiler design, with experience in optimizing performance for complex models. Your expertise in Python and familiarity with frameworks like TensorFlow, PyTorch, and JAX enable you to transform ML models effectively for deployment on AWS Inferentia and Trainium. You are skilled in solving challenging compiler optimization problems, ensuring that various ML model families achieve optimum performance. Your technical communication skills allow you to collaborate effectively with internal and external stakeholders, driving the success of the Neuron compiler.
You understand the intricacies of large language models and other advanced ML architectures, which helps you make informed decisions about compiler implementations. Your experience in object-oriented programming languages equips you with the tools needed to tackle complex software challenges. You are passionate about contributing to the AI revolution and democratizing access to AI technologies.
What you'll do
In this role, you will be responsible for building the next generation of the Neuron compiler, which is crucial for optimizing ML models for AWS's custom chips. You will work closely with various ML frameworks to ensure seamless deployment and performance enhancement. Your responsibilities will include solving hard compiler optimization problems and collaborating with teams to bring new products and features to market. You will also engage in pre-silicon design discussions, ensuring that the Neuron compiler is both high-performing and user-friendly.
You will leverage your understanding of how different ML models operate to coax the compiler into generating optimal implementation instructions. Your role will involve continuous learning and adaptation as you work on cutting-edge technologies in the AI space. You will be part of a dynamic team that is dedicated to making deep learning accessible to everyday developers.
What we offer
Amazon provides a competitive salary range of $165,200.00 - $223,600.00 annually, along with comprehensive benefits including health insurance, retirement plans, and paid time off. You will have the opportunity to work in a collaborative environment that fosters innovation and growth. Join us in shaping the future of AI and deep 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 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.
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.
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.