Senior AI Hardware Systems Engineer, Annapurna Labs, Trainium Machine Learning Fleet Operations

Description
Annapurna Labs designs silicon and software that accelerates innovation. Customers choose us to create cloud solutions that solve challenges that were unimaginable a short time ago—even 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. The Machine Learning Acceleration Fleet Operations Team is looking for candidates interested in diving deep into our fleet of ML servers deployed around the world.
We are seeking an engineer who is comfortable debugging emergent problems in GPU and server hardware, writing scripts in languages such as Python or Bash, running large scale experiments on a fleet of complex hardware, developing data infrastructure and analyzing trends, and building automation software to scale operations.
Our team has end to end ownership of some of the most advanced server hardware in the world. We drive technical debug efforts and write truly massive scale autonomous software to monitor, optimize, and remediate machine learning hardware. Come join us!
Key job responsibilities
- Member of a team responsible for system remediation, operational excellence, and customer experience on bleeding edge ML products
- Utilize data to root cause hardware failures and identify live trends on the most complex systems in AWS
- Implement and improve system level testing across the product lifecycle
- Develop software which can be maintained, improved upon, documented, tested, and reused
- Dive deep on issues at the intersection of hardware and software
A day in the life
As a Platform Development Engineer, you are the dedicated owner of an ML server platform in our fleet. Your mission is to maximize its health, sellability, and customer experience.
You start each day with eyes on the fleet — reviewing dashboards to identify trends and triaging emergent issues, then partnering with hardware and software engineering teams to debug, investigate, and translate findings into permanent fixes. You own the end-to-end testing story and manage tradeoffs between coverage and velocity. You direct new automations, tooling, and data infrastructure to scale your operations. You manage software deployments, debug issues with them, and run status meetings to align all platform stakeholders on how the product is performing.
About the team
The MLA Fleet Operations team was formed to maintain an exceptionally high quality bar for our fleet of advanced machine learning accelerators and server products. We perfect the customer experience by developing scalable software for rapid incident response times and data visualization as well as diving deep into hardware issues as they arise.
Basic Qualifications
- 4+ years of programming with at least one modern language such as C++, C#, Java, Python, Golang, PowerShell, Ruby experience
- 3+ years of non-internship professional software development experience, or Bachelor's degree or above in engineering or equivalent
- 3+ years of designing or architecting (design patterns, reliability and scaling) of new and existing systems experience
- Experience in computer architecture, or experience with general troubleshooting/debugging of hardware
- Experience working with device technologies under development, familiarity with flashing firmware, basic device debugging and familiarity with reading/pulling device logs
- BS degree in computer science, computer engineering, or related field, or 4+ years of technical work experience
- 2+ years of server hardware troubleshooting and repair experience
Preferred Qualifications
- 7+ years of leading design or architecture (design patterns, reliability and scaling) of new and existing systems experience
- Experience with concepts such as system architecture, optimization, system dynamics, system analysis, statistical analysis, reliability analysis, and decision making
- Experience that includes strong analytical skills, attention to detail, and effective communication abilities, or experience troubleshooting and debugging technical systems and experience with automation and any version control tools
- Knowledge of operating systems, hardware, storage, network, security, database administration and cloud infrastructure
- Master's degree or above in electrical engineering, computer engineering, or equivalent
- Experience with SOC bring-up and post-silicon validation
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 https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t 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 https://amazon.jobs/en/benefits.
USA, TX, Austin - 159,200.00 - 215,300.00 USD annually