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 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 .

USA, CA, Cupertino - 188,900.00 - 330,500.00 USD annually