Ml Compiler Engineer • Mid-Level

Who you are

You have 3+ years of experience in compiler development, with a strong background in designing and architecting systems for reliability and scaling. Your expertise in C++ and Python is critical, as you will be developing tooling for the Neuron Compiler's accuracy and reliability. A science background in compiler development is preferred, and while a background in Machine Learning and AI accelerators is not required, it is a plus. You are passionate about pushing the boundaries of technology and enjoy working in a collaborative team environment.

What you'll do

As a ML Compiler Engineer II on the Neuron Compiler Automated Reasoning Group, you will be responsible for developing and maintaining tooling that enhances the Neuron Compiler's capabilities. You will lead efforts in building fuzzers and specification synthesis tooling for our LLVM-based compiler, ensuring that our technology remains at the forefront of innovation. You will collaborate closely with teams focused on Machine Learning accelerators, contributing to the optimization of application models across various domains, including Large Language and Vision. Your role will involve engaging with leading frameworks such as PyTorch, TensorFlow, and JAX, and you will strive to deliver high-quality solutions that meet customer needs.

What we offer

At Amazon, we provide a comprehensive benefits package that includes competitive compensation, equity options, and a full range of medical and financial benefits. You will have the opportunity to work in a dynamic environment that encourages innovation and professional growth. We value diversity and encourage you to apply even if your experience doesn't match every requirement. Join us in shaping the future of Machine Learning technology.