SoC Platform Software Engineering Manager, Annapurna Labs Machine Learning Acceleration, AWS
Cupertino, CA - USA
Department:
Job Summary
Our SoC HAL (Hardware Abstraction Layer) team builds the platform software layer for AWSs custom Trainium and Inferentia ML accelerator chips. The HAL is a shared library that boots configures and manages every hardware block on the SoC 270 instances per chip and the same source tree compiles and runs on SystemVerilog DPI for chip verification QEMU for system emulation and Carbon OS in microcontrollers within the AWS production fleet. Your platform abstractions are what make this possible and your APIs are the interface that 100s of engineers across verification emulation and production use to interact with the chip.
Tech stack: C17 CMake GoogleTest Python SystemVerilog DPI SPI APB/AXI bus protocols PCIe UCIe HBM PLL custom IPs
As the SoC Platform Software Manager you will:
- Manage coach and grow a team of 6 engineers set technical direction own hiring and create an environment where strong engineers want to stay
- Own the platform abstraction layer that enables one C codebase to compile and run correctly across three target environments with fundamentally different runtime characteristics
- Shape the external API contracts that verification emulation and production teams build on balancing stability for consumers against the need to evolve as new chip generations arrive
- Drive the architecture of our C template metaprogramming framework that generates type-safe register interfaces for every hardware block and our BUTR (Built-in Unit Test for Registers) and HITL (Hardware-in-the-Loop) test infrastructure
- Build and maintain the CI/CD and validation strategy that catches integration issues across all three platforms before they reach customers
- Coordinate across chip architects RTL designers verification engineers validation engineers and platform software teams youre the single point of accountability for HAL readiness on every new chip program
- Get into the weeds alongside your team debug register-level HW/SW interactions review code and write code yourself when it matters
Most platform software teams target one OS or one hardware family. We target three execution environments from a single source tree and our software must be stateless survive live-updates on running production servers without reboots and be correct down to individual register bits. A single abstraction leak can break chip verification stall emulation or misconfigure millions of servers in AWSs global fleet.
The HAL runs on an external microcontroller running embedded Linux reaching into the chip over SPI and PCIe. Its stateless by design: the microcontroller can reboot at any time including during customer workloads and the HAL must resume managing the SoC by querying hardware state on-demand. Your platform layer is what makes this resilience possible while keeping the complexity invisible to consumers.
The same codebase that runs in pre-silicon simulation months before tape-out is the codebase that runs in production fleet. When the chip comes back from the fab your team validates that pre-silicon models match real hardware behavior. For Trainium3 our HAL enabled a full ML training workload within 12 hours of first power-on: ML background needed. Your platform software is the foundation that enables ML training across clusters of thousands of interconnected accelerators youll work on components like PCIe and HBM but wont need to understand ML itself.
This role can be based in Cupertino CA or Austin TX. The team is split between the two sites.
- 3 years of engineering team management experience
- 7 years of professional software development in C or C including systems platform or infrastructure software
- 4 years of designing or architecting software systems (platform abstractions API design multi-target build systems)
- Experience developing software that interfaces with hardware or runs across multiple execution environments
- Experience designing APIs or abstraction layers consumed by other engineering teams
- Experience in recruiting hiring mentoring/coaching and managing teams of Software Engineers to improve their skills and make them more effective product software engineers
- Experience building or maintaining hardware abstraction layers board support packages or platform software for SoC ASIC or embedded systems
- Experience with multi-platform or cross-compilation build systems (targeting simulation emulation and production from a single source tree)
- Familiarity with bus protocols (APB AXI PCIe) or memory subsystems (HBM DDR)
- Experience with C template metaprogramming or code generation frameworks
- Experience with pre-silicon software development (simulation emulation or virtual platforms)
Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status disability or other legally protected status.
Los Angeles County applicants: Job duties for this position include: work safely and cooperatively with other employees supervisors and staff; adhere to standards of excellence despite stressful conditions; communicate effectively and respectfully with employees supervisors and staff to ensure exceptional customer service; and follow all federal state and local laws and Company policies. Criminal history may have a direct adverse and negative relationship with some of the material job duties of this position. These include the duties and responsibilities listed above as well as the abilities to adhere to company policies exercise sound judgment effectively manage stress and work safely and respectfully with others exhibit trustworthiness and professionalism and safeguard business operations and the Companys reputation. Pursuant to the Los Angeles County Fair Chance Ordinance we will consider for employment qualified applicants with arrest and conviction records.
Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process including support for the interview or onboarding process please visit for more information. If the country/region youre applying in isnt listed please contact your Recruiting Partner.
The base salary range for this position is listed below. Your Amazon package will include sign-on payments and restricted stock units (RSUs). Final compensation will be determined based on factors including experience qualifications and location. Amazon also offers comprehensive benefits including health insurance (medical dental vision prescription Basic Life & AD&D insurance and option for Supplemental life plans EAP Mental Health Support Medical Advice Line Flexible Spending Accounts Adoption and Surrogacy Reimbursement coverage) 401(k) matching paid time off and parental leave. Learn more about our benefits at CA Cupertino - 212700.00 - 287700.00 USD annually
USA TX Austin - 184900.00 - 250200.00 USD annually
Required Experience:
Manager
About Company
Free shipping on millions of items. Get the best of Shopping and Entertainment with Prime. Enjoy low prices and great deals on the largest selection of everyday essentials and other products, including fashion, home, beauty, electronics, Alexa Devices, sporting goods, toys, automotive ... View more