Principal Developer Advocate – AWS Trainium, Trainium

Do you love getting hands-on with ML hardware acceleration and sharing your passion for open-

source AI with the broader developer community?

Amazon Web Services (AWS) is looking for a Principal Developer Advocate to join the

Annapurna Labs team. AWS Machine Learning accelerators are at the forefront of AWS

innovation. The Inferentia chip delivers best-in-class ML inference performance at the lowest

cost in cloud. Trainium will deliver the best-in-class ML training performance with the most

teraflops (TFLOPS) of compute power for ML in the cloud. This is all enabled by our software

stack, the AWS Neuron Software Development Kit (SDK), which includes an ML compiler,

runtime and natively integrates into popular ML frameworks including PyTorch. In this role, you

will be the external face of AWS in ML hardware acceleration, combining deep technical

expertise, strategic vision, and authentic community engagement to drive awareness, trust, and

adoption of Trainium and the Neuron SDK across the global ML developer ecosystem.

Ideally, you come from a software engineering, ML research, or systems performance

background - but you love being on stage, on screen, or in the open-source community guiding

developers toward technologies you believe in. You have immediate credibility with ML

practitioners, researchers, and performance engineers. You are excited about open-source AI,

hardware-software co-design, and the future of ML infrastructure. You exhibit strong judgment

in selecting strategic opportunities and can operate with full independence across multiple

technical communities

Key job responsibilities

Creating and maintaining relevant content (demos, videos, blogs, podcasts, livestreams, etc),

covering AWS Trainium and the Neuron SDK along with emerging technologies and industry

developments.

Gaining industry recognition and credibility as a regular attendee and speaker at third party

technology conferences, shows and press events.

Curating, developing and delivering compelling presentations to help drive awareness and

adoption of AWS Trainium.

Maintaining a credible presence in social media

Ability to travel (including internationally when needed) and ability to work autonomously

A day in the life

As a technical champion, you'll create and deliver content that makes Trainium and the Neuron

SDK accessible, credible, and compelling to ML developers worldwide - from getting-started

tutorials to deep-dive kernel optimization guides. As a community builder, you'll engage with

ML practitioners, researchers, and performance engineers to understand their challenges,

represent their voice internally, and translate community insights into product roadmap impact.

As an ecosystem strategist, you'll build long-term partnerships with key open-source projects and

communities, co-develop integrations, and ensure Trainium is a visible and credible option

wherever ML developers make hardware decisions.

About the team

The Amazon Annapurna Labs team is a responsible for building innovation in silicon and

software for AWS customers. We are at the forefront of innovation by combining cloud scale

with the world’s most talented engineers. Our team covers multiple disciplines including silicon

engineering, hardware design and verification, software and operations. Because of our teams

breadth of talent, we have been able to improve AWS cloud infrastructure in networking and

security with products such as AWS Nitro, Enhanced Network Adapter (ENA), and Elastic

Fabric Adapter (EFA), in compute with AWS Graviton and the EC2 F1 FPGA instances, in

storage with scalable NVMe, and now in AI and Machine Learning with AWS Neuron SDK,

Inferentia and Trainium ML accelerators.

Basic qualifications

- Experience conveying complex technical concepts to both technical and business audiences

- Experience developing technology solutions and evangelising end-to-end technology roadmaps that guide IT transformations toward cloud computing

- Experience communicating complex ideas to technical and non-technical audiences

- 10+ years of Experience in software engineering, ML research, systems performance, or a related technical field

- Experience with machine learning frameworks, tools, or infrastructure

- Ability to work autonomously and manage multiple initiatives across different technical communities

- Experience creating and delivering presentations, workshops, or demonstrations to technical and non-technical audiences

- Proficiency with at least one major programming language or technology platform

Preferred qualifications

- Experience speaking at industry conferences, user groups, or other public forums

- Background in software development, cloud architecture, or enterprise technology solutions

- Familiarity with content creation tools and platforms, including video production and social media

- Experience working with open-source communities or contributing to open-source projects

- Knowledge of emerging technology trends and their potential business applications

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 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, CA, Cupertino - 188,900.00 - 330,500.00 USD annually