Senior Device Driver Engineer (Team Lead), Annapurna Labs Machine Learning Accelerators, AWS

AWS Utility Computing (UC) provides product innovations that continue to set AWS's services and features apart in the industry. As a member of the UC organization, you'll support the development and management of Compute, Database, Storage, Platform, and Productivity Apps services in AWS, including support for customers who require specialized security solutions for their cloud services. This role may involve exposure to and experience with Amazon's growing suite of generative AI services and other cutting-edge cloud computing offerings across the AWS portfolio.

Annapurna Labs (our organization within AWS UC) designs silicon and software that accelerates innovation. We're looking for someone to lead our SoC (System on Chip) device-driver / HAL (Hardware Abstraction Layer) software team. You'll help us deliver at scale, as we build the next generation of SoC software. This is a hands-on, in-the-trenches software engineering leadership position.

As the team lead for SoC driver SW, you will:

• Build and manage a small, strong team of 3-5 developers
• Work with hardware designers to write drivers for newly developed SoC IPs
• Work with system software teams to solve SoC and system-level architectural issues, drive debug, and innovate on cross-functional solutions
• Refactor and maintain existing codebases throughout the device lifecycle
• Continuously test and deploy your software stack to multiple internal customers
• Innovate on the tooling you provide to customers, making it easier for them to use and debug our SoCs

You will thrive in this role if you:

• Enjoy building, managing, and leading small teams
• Love solving complex system-level issues
• Are proficient in C++ and familiar with Python
• Know how to build effective abstractions over low-level SoC details
• Are familiar with modular driver architectures (such as the Linux or Windows driver stacks)
• Have strong opinions about software architecture, and are able to apply them effectively
• Enjoy learning new technologies, building software at scale, moving fast, and working closely with colleagues as part of a small, startup-like team within a large organization

This role can be based in either Cupertino, CA or Austin, TX. The team is split between the two sites, with no preference for one over the other.

This is a fast-paced role where you'll work with thought-leaders in multiple technology areas. You'll have high standards for yourself and everyone you work with, and you'll be constantly looking for ways to improve your software, as well as our products' overall performance, quality, and cost.

We're changing an industry. We're searching for individuals who are ready for this challenge, who want to reach beyond what is possible today. Come join us and build the future of machine learning!