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 an ML Kernel Performance Engineer to work on AWS Neuron, focusing on optimizing performance for deep learning and GenAI workloads. You'll leverage your expertise in AWS and machine learning frameworks like TensorFlow and PyTorch. This role is based in Toronto.
Job Description
Who you are
You have a strong background in machine learning and high-performance computing, with experience in developing software for custom hardware accelerators. Your expertise in AWS and familiarity with deep learning frameworks like TensorFlow and PyTorch enable you to craft high-performance kernels that maximize efficiency. You understand the intricacies of working at the hardware-software boundary and are passionate about pushing the limits of AI acceleration.
You possess a deep understanding of performance optimization techniques and are skilled in analyzing and improving the efficiency of machine learning models. Your collaborative nature allows you to work effectively with cross-functional teams, ensuring that the solutions you develop meet the needs of customers and align with future architectural designs.
What you'll do
In this role, you will be responsible for developing and optimizing high-performance kernels for AWS's custom ML accelerators, Inferentia and Trainium. You will work closely with the Acceleration Kernel Library team to ensure that every FLOP counts in delivering optimal performance for demanding workloads. Your work will involve integrating the AWS Neuron SDK with popular ML frameworks, enabling seamless performance enhancements for deep learning and GenAI applications.
You will engage with customers to understand their specific needs and challenges, using this feedback to inform your kernel development and optimization strategies. Additionally, you will contribute to the future architecture designs of the AWS Neuron SDK, ensuring that it remains at the forefront of AI acceleration technology.
What we offer
Amazon provides a comprehensive benefits package, including medical, dental, and vision insurance, as well as a Registered Retirement Savings Plan (RRSP) and a Deferred Profit Sharing Plan (DPSP). You will also enjoy paid time off and resources aimed at improving your health and well-being. The salary for this position ranges from CAD 114,800 to CAD 191,800 annually, reflecting the importance of this role within the organization.
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
Ml Compiler Engineer
Amazon is hiring an ML Compiler Engineer for the Annapurna Labs team to develop the AWS Neuron SDK, which accelerates deep learning workloads. You'll work with technologies like TensorFlow and PyTorch to optimize performance on custom ML accelerators. This role requires expertise in machine learning frameworks.
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.
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 Senior Machine Learning Kernel Performance Engineer to work on AWS Neuron, focusing on optimizing performance for deep learning and GenAI workloads. You'll leverage your expertise in AWS, Python, and PyTorch to enhance machine learning accelerators. This role requires strong experience in high-performance computing.
Software Development Manager
Amazon is hiring a Senior Software Development Manager for AWS Neuron to lead the development of cutting-edge machine learning accelerators. You'll work with technologies like AWS, PyTorch, and TensorFlow. This position requires significant experience in software development and team leadership.