Sr. GPU/Accelerator Hardware Development Engineer, Annapurna Labs
About The Position
This role is part of the AWS Trainium Machine Learning Acceleration team, focusing on the system design and optimization of hardware in data centers. The position involves leading bleeding-edge hardware development projects for next-generation AI accelerator compute systems, specifically for AWS's Project Rainier, a significant AI infrastructure initiative. The team develops a complete vertical stack system, from silicon to hardware to software, and deploys it directly to customers in their own data centers. The role requires providing leadership in applying new technologies to large-scale server deployments to enhance customer experience. It's a fast-paced, intellectually challenging position working with thought leaders, demanding high standards for performance, quality, and cost, with the goal of changing the industry.
Requirements
- 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.
- Strong knowledge in electrical engineering fundamentals, power & signal integrity, and analog/digital circuits
- 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
Nice To Haves
- Lead bleeding-edge HW development projects
- Experience with AWS (Amazon Web Services)
- Experience with Anthropic’s Claude AI models
- Experience with Trainium2 chips
- Experience working in a fast-paced start-up like environment
- Ability to work independently to deliver multiple projects in parallel
- Highly motivated and detailed oriented
- Meeting the highest standards and time to market, cost and quality goals
Responsibilities
- System design and optimization of hardware in AWS data centers.
- Provide leadership in the application of new technologies to large scale server deployments.
- System design, validation, and integration of hardware in the AWS fleet through its entire life cycle.
- 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.
- Drive ODM HW development and testing and be part of the Production flow definition team.
- Drive component selection and validation of electrical, mechanical components, cables.
- 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.
Benefits
- 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
- parental leave
- sign-on payments
- restricted stock units (RSUs)
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Senior
Education Level
Associate degree
