Device Driver Engineer, Annapurna Labs Machine Learning Accelerators

AWS is seeking skilled engineers to join their Annapurna Labs team, focusing on developing drivers for cutting-edge machine learning accelerator chips. This role involves working at the heart of AWS Machine Learning servers (TRN and INF), creating and maintaining the critical low-level infrastructure that powers AWS's machine learning capabilities.

The position offers an exciting opportunity to work with custom silicon chips and develop essential driver software that ensures chip functionality, performance, and security. You'll be part of a dynamic team split between Cupertino, CA and Austin, TX, working in a fast-paced environment alongside thought-leaders in multiple technology areas.

As a Device Driver Engineer, you'll collaborate closely with hardware designers and system software teams, building and maintaining the foundational software that manages AWS's custom system-on-chips (SoCs). The role requires strong C++ programming skills, understanding of driver architectures, and the ability to work with hardware-based systems.

What makes this role unique is its position at the intersection of hardware and software in AWS's machine learning infrastructure. While no machine learning background is required, you'll be developing crucial components that enable AWS's machine learning capabilities. The team operates with a startup-like atmosphere within Amazon's larger organization, offering the best of both worlds - the agility of a small team with the resources and impact of a tech giant.

The compensation is highly competitive, ranging from $129,300 to $223,600 based on location and experience, plus additional benefits including equity and sign-on payments. AWS values diversity and work-life harmony, offering an inclusive environment with strong mentorship opportunities and clear paths for career growth.