Machine Learning Engineer • Entry-Level

Who you are

You are a recent graduate or early career professional eager to dive into the world of machine learning and silicon innovation. You have a foundational understanding of AI principles and are excited about the opportunity to work on groundbreaking projects that push the boundaries of technology. Your passion for hardware design, distributed systems, and compiler development sets you apart as a bold innovator ready to contribute to the future of cloud computing.

You possess skills in programming languages such as Python and are familiar with machine learning frameworks like PyTorch and JAX. You are eager to learn and grow in a collaborative environment, where you can apply your knowledge to real-world challenges. Your ability to work in teams and communicate effectively will help you thrive in this dynamic role.

What you'll do

As a Machine Learning Engineer at Annapurna Labs, you will engage in projects that directly impact millions of AWS customers worldwide. You will have the opportunity to work on architecting next-generation AI chips that process billions of parameters, building distributed systems that scale across thousands of accelerators, and creating compiler optimizations that enhance ML training speed. You will collaborate with a diverse team of engineers and researchers, contributing to the development of custom silicon and revolutionary software systems.

Your responsibilities will include optimizing machine learning frameworks, developing performance engineering solutions, and contributing to ML infrastructure development. You will be involved in hands-on coding, testing, and deploying machine learning models, ensuring they meet the high standards expected at AWS. You will also participate in brainstorming sessions and technical discussions, where your ideas and insights will be valued as part of the innovation process.

What we offer

At Amazon, we provide a supportive environment for early career professionals to grow and develop their skills. You will have access to mentorship from experienced engineers and opportunities to work on cutting-edge technology that shapes the future of AI. Our benefits include competitive salaries, 401(k) matching, paid time off, and parental leave. Join us at Annapurna Labs and be part of a team that is not just participating in the AI revolution but accelerating it.