Memory Subsystem Engineer, MLA Technology

Annapurna Labs (our organization within AWS UC) designs silicon and software that accelerates innovation. Customers choose us to create cloud solutions that solve challenges that were unimaginable a short time agoeven yesterday. Our custom chips accelerators and software stacks enable us to take on technical challenges that have never been seen before and deliver results that help our customers change the world.

In Annapurna Labs we are at the forefront of hardware/software co-design not just in Amazon Web Services (AWS) but across the industry. Our Machine Learning Accelerator (MLA) Technology is seeking a memory subsystem engineer who is interested in diving deep into the definition design validation and data center operation of AWSs next generation machine learning silicon and servers. Our team spans HBM SRAM and chiplet interconnects to enable the highest performing memory subsytems in the world.

As a senior member of our technology team you will have opportunities to participate in the design and execution of HBM SRAM UCIe and general high speed analog technologies with the goal of creating the most stable machine learning platforms within AWSs data centers. A senior memory subsytem engineer on our team needs to be able to work with vendors and internal design teams understand complex memory and SERDES features write/modify tests at scale debug fleet wide issues and collect data from manufacturing and the data center.

Our broader team has end to end ownership of some of the most complicated IPs on the most advanced server hardware in the world. We drive complex technical debug efforts involving our IPs and leverage the massive scale of EC2 to monitor optimize and improve our machine learning hardware reliability on behalf of our customers.


Key job responsibilities
As a senior member of the team you will join a mixed group of hardware and software engineers working to design integrate and innovate the next generation of machine learning chips into Trainium this position it is expected that you will:

* Collaborate with architects design teams and software engineers on our next generation ML chips
* Support on-going debug and operations of previous ML chips within manufacturing and the data center
* Dive deep into IP integration packaging silicon bring up characterization and validation of our broader memory subsystems
* Independently develop the scripts you need to execute and collaborate with software engineers as your needs scale



A day in the life
A day in the life of a CHDE focused on memory subsystems on the MLA Technology team focuses on operational excellence constructively identifying problems prototyping solutions and leading data collection at scale to improve our products. We start each day looking at our fleet reviewing dashboards for emergent issues impacting our customers partnering with other teams to drive complex debugs as it pertains to UCIe and associated SoC subsystems. We then look forward to the future technologies being developed and how we can best focus our efforts to help improve them and ensure a high quality product on behalf of our customers.

Our team members touch everything from electrical simulations to hardware qualification on test benches to software driven data center metrics with a broad range of tasks across multiple skillsets where you can help improve the reliability and performance of our products. You help the team evolve by actively participating in design discussions team planning code reviews tickets/metric reviews and data center capacity initiatives. CHDEs on the MLA Technology Team are expected to help mentor others on the team in their area of expertise to help develop the teams baseline skillsets and to participate in the hiring process for the team.


- Bachelors degree in Electrical Engineering Computer Engineering Systems Engineering or related fields
- 5 years of practical semiconductor design work including full-chip and subsystem integration experience
- Good knowledge of memory training timing parameters and/or controller features
- Ability to create scripts (lua bash python etc.) to accomplish functional day to day tasks.
- Knowledge of DDR/HBM/UCIe phy and controller related protocols

- Drive cross-functional triage effort on functional and performance issues
- 3 years in SOC/IO/Subsystems Experience working closely with physical design teams to develop highly optimized ASICs with excellent power performance and area
- Support the physical design team with IP integration 2.5D packaging clocking and timing constraints
- Drive cross-functional triage effort on functional and performance issues
- Perform system-level debug and root-cause analysis through bring-up characterization validation and production phases

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status disability or other legally protected status.

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 TX Austin - 159200.00 - 215300.00 USD annually