Machine Learning Engineer • Entry-Level

Who you are

You are a bold innovator eager to contribute to the AI revolution — your passion for technology drives you to explore the boundaries of what's possible in machine learning and silicon innovation. You have a foundational understanding of machine learning principles and are excited to apply your knowledge in real-world scenarios. You are comfortable working with programming languages like Python and are familiar with frameworks such as PyTorch and JAX, which are essential for optimizing machine learning systems.

You thrive in collaborative environments and are eager to learn from experienced professionals — your ability to communicate effectively will help you engage with cross-functional teams, including hardware designers and software engineers. You are motivated by the opportunity to tackle complex technical challenges and contribute to projects that impact millions of AWS customers worldwide.

What you'll do

As an intern at Annapurna Labs, you will work on cutting-edge projects that involve designing custom silicon and developing revolutionary software systems. You will have the chance to contribute to various technical tracks, including ML Systems & Compilers and Systems Software & Infrastructure. Your responsibilities may include optimizing machine learning frameworks, developing compiler optimizations, and building distributed training systems that scale across thousands of accelerators.

You will collaborate with a team of talented engineers and researchers, gaining hands-on experience in the field of AI acceleration. Your work will directly influence the performance and efficiency of machine learning models, helping to shape the future of cloud computing at AWS. You will also have opportunities to participate in brainstorming sessions and contribute your ideas to innovative solutions.

What we offer

Amazon provides a dynamic internship experience where you can make meaningful contributions from day one. You will receive mentorship from industry experts and have access to resources that will help you grow your skills in machine learning and silicon design. The internship program includes competitive compensation and benefits, as well as opportunities for networking and professional development within the company. Join us at Annapurna Labs and be part of a team that is inventing the future of AI infrastructure.