Software Engineer - Hardware Abstraction Layer, AWS Machine Learning Accelerators
Description For Software Engineer - Hardware Abstraction Layer, AWS Machine Learning Accelerators
AWS is seeking talented engineers to join their system software team developing the hardware abstraction layer (HAL) for cutting-edge machine learning accelerators. This role is part of AWS's Annapurna Labs organization, which designs and deploys some of the largest custom silicon in the world. The position focuses on developing critical infrastructure management software for AWS's machine learning servers (Trainium and Inferentia).
The role requires expertise in C++ programming and hardware systems, working at the lowest level of AWS infrastructure to ensure chip functionality, performance, and security. You'll collaborate with hardware designers and system software teams to build HALs for new SoC IPs, solve architectural challenges, and innovate on cross-functional solutions.
This is an exciting opportunity to work with thought-leaders in multiple technology areas, building the future of machine learning infrastructure. The team operates with a startup-like mentality within AWS, focusing on fast-paced development and innovation. While the role involves working with ML chips, no machine learning background is required as the focus is on low-level system software development.
The position offers competitive compensation, comprehensive benefits, and significant growth opportunities. AWS values diversity, work-life harmony, and maintains an inclusive culture with various employee-led affinity groups. The team supports knowledge-sharing and mentorship, making it an ideal environment for professional development.
Based in either Cupertino, CA or Austin, TX, you'll be part of a team split between both locations, working on challenging problems that push the boundaries of what's possible in cloud computing and machine learning acceleration.
Responsibilities For Software Engineer - Hardware Abstraction Layer, AWS Machine Learning Accelerators
- Work with hardware designers to build HALs for newly developed SoC IPs
- Work with system software teams to solve SoC and system-level architectural issues
- Drive debug, architect the HAL, and innovate on cross-functional solutions
- Continuously test and deploy software stack to multiple internal customers
- Refactor and maintain existing codebases throughout the device lifecycle
- Innovate on tooling provided to customers
Requirements For Software Engineer - Hardware Abstraction Layer, AWS Machine Learning Accelerators
- 5+ years of non-internship professional software development experience
- 5+ years of non-internship design or architecture experience
- Experience programming with C++
- Knowledge of modular driver architectures
- Strong software architecture skills
- Experience with hardware-based systems
Benefits For Software Engineer - Hardware Abstraction Layer, AWS Machine Learning Accelerators
- Medical benefits
- Financial benefits
- Equity compensation
- Sign-on payments
- Mentorship and career growth opportunities
- Inclusive team culture
- Work-life harmony
