Machine Learning Engineer • Senior

Who you are

You have 5+ years of experience in machine learning and compiler engineering, 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 machine learning accelerators. You thrive in collaborative environments and enjoy tackling complex technical challenges.

You possess a deep understanding of AWS services and how they integrate with machine learning workflows. Your experience includes developing and optimizing compilers, and you have a solid grasp of the underlying hardware architecture that supports machine learning applications. You are passionate about advancing the state of the art in machine learning and are eager to contribute to innovative projects.

What you'll do

As a Senior Machine Learning Engineer at Amazon, you will be responsible for developing and optimizing the AWS Neuron Software Development Kit (SDK), which includes a deep learning compiler and runtime. You will work closely with silicon engineers to ensure that the compiler effectively translates neural network descriptions into efficient executable code. Your role will involve collaborating with cross-functional teams to enhance the performance of AWS Machine Learning accelerators.

You will lead efforts to integrate the compiler with popular machine learning frameworks, ensuring seamless functionality and performance improvements. Your work will directly impact the efficiency of machine learning workloads for AWS customers, enabling them to leverage cutting-edge technology for their applications. You will also mentor junior engineers, sharing your knowledge and fostering a culture of innovation within the team.

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 have the opportunity to work in a dynamic environment at the forefront of machine learning technology, contributing to projects that have a significant impact on the industry. We encourage you to apply even if your experience doesn't match every requirement, as we value diverse perspectives and backgrounds.