Sr. GPU/Accelerator Hardware Development Engineer, Annapurna Labs

DESCRIPTION

Would you like to develop the Next Generation of AI accelerator compute systems? Lead bleeding-edge HW development projects? Have you heard of Amazon Web Services (AWS) Project Rainer? This is the opportunity to be a part of a fast-moving innovation team that is changing the world of AI at massive scale. At AWS Trainium we develop a complete vertical stack system, from our own Silicon to Hardware to Software and deploy directly to our customers in our own Data Centers

We are seeking experienced Lead System Design Engineers to build the next generation of our cloud server infrastructure, Project Rainier. Project Rainier is a massive $11 billion Amazon Web Services (AWS) AI infrastructure initiative, featuring one of the world's largest compute clusters dedicated to training and running Anthropic’s Claude AI models. It utilizes over 500,000 custom Trainium2 chips, designed for high-performance AI training.

As a member of the AWS Trainium Machine Learning Acceleration team you’ll be responsible for the System design and optimization of hardware in our data centers. You’ll provide leadership in the application of new technologies to large scale server deployments in a continuous effort to deliver a world-class customer experience. This is a fast-paced, intellectually challenging position, and 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 products performance, quality and cost. We’re changing industry, and we want individuals who are ready for this challenge and want to reach beyond what is possible today.

Key job responsibilities

We are looking for a Lead Hardware Design Engineer with strong skills in both hardware and software. In this role, you will be responsible for system design, validation, and integration of hardware in the AWS fleet through its entire life cycle. You will work cross functionally with AWS monitoring teams, members of the Hardware Design team, and

additional teams across AWS to improve quality and reliability of products operating in the fleet.

We are looking for candidates who thrive in a fast-paced start-up like environment and work independently to deliver multiple projects in parallel. To be successful, you need to be highly motivated and detailed oriented while meeting the highest standards and time to market, cost and quality goals.

BASIC QUALIFICATIONS

• BS or MS degree in Electrical or Computer Engineering (EE / CE)

• Minimum of 5 years of experience with High-Speed system design and validation

• Experience with Schematic and layout tools.

• Drive ODM HW development and testing and be part of the Production flow definition team

• Strong knowledge in electrical engineering fundamentals, power & signal integrity, and analog/digital circuits

• Able to drive component selection and validation of electrical, mechanical components, cables

• Experience with hardware development process and system development across full product life cycles

• Experience using lab equipment such as bench power supplies, high-speed oscilloscopes, logic analyzers, spectrum analyzers, VNA’s, and thermal chambers

• Experience with supply chain management

PREFERRED QUALIFICATIONS

• Lead end-to-end server hardware development lifecycle from Concept, Architecture, Design, Validation and Production

• Drive PCB board design for server motherboards, accelerator carrier boards, and high-speed interconnect boards.

• Collaborate with silicon, firmware, and system software teams to enable optimal hardware/software co-design.

• Improve compute density, power efficiency, and network bandwidth utilization.

• Drive root cause analysis for hardware issues during validation and production.

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 Company’s 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 Ama