AWS Trainium is deployed at scale, with millions of chips in production, and has been used for training and inference of frontier models. AWS Neuron is the software stack for Trainium, enabling customers to run deep learning and generative AI workloads with optimal performance and cost efficiency.
AWS Neuron is hiring a Senior Technical Writer to drive the documentation strategy and AI-based content contribution and automation initiatives for our AWS Neuron SDK, a software development kit that enables machine learning with our specialized AWS Inferentia and Trainium chips.
You will combine your passion for technical writing and your strategic savvy to design mechanisms and web experiences that incorporate AI to enable Neuron customers and documentation contributors alike to deliver and use professional, consistent technical, AI-ready documentation experiences. Working at the intersection of machine learning, hardware acceleration, and developer experience, you'll lead efforts to streamline our documentation processes while ensuring our docs site and technical content meets the high standards that AWS customers expect.
To be successful in this role, you will partner with multiple Neuron component engineering teams, PMs responsible for Neuron compiler, runtime, and NKI, as well as Solution Architects supporting customers. You will develop deep knowledge of documentation design and organization, information architecture, AI agents and LLMs, and the Neuron SDK to effectively define documentation strategy and make informed content decisions.

The Ideal Candidate:

The ideal candidate can balance competing priorities across multiple workstreams and drive alignment across engineering and business stakeholders in a fast-moving, early-stage product environment, with excellent written and verbal communication abilities. The ideal candidate has experience with:

• Technical writing and documentation management for developer-facing products
• Python and Git-based code tools development
• Documentation design and organization, information architecture, and web site development
• Product strategy and feature design for documentation experiences
• LLM-based tools and AI agents for content automation
• Interactive, code-based documentation experiences

About AWS Neuron:

AWS Neuron is the software stack for running deep learning and generative AI workloads on AWS Trainium and AWS Inferentia. It includes a compiler, runtime, training and inference libraries, and developer tools for monitoring, profiling, and debugging. Built on an open source foundation, Neuron supports native Py Torch and JAX frameworks and popular ML libraries without code modification. Neuron enables rapid experimentation, distributed training across multiple chips and nodes, and cost-optimized inference powered by optimized kernels.
Key job responsibilities
Content Development & Delivery: Work with multiple Neuron component engineering teams to plan content for Neuron releases and drive the delivery of the materials. Develop content that is based on code and provides interactive experiences for Neuron's developer users. Manage and report on documentation efforts and progress in an Agile environment.
Documentation Strategy & Vision: Define and deliver strategy for the delivery of effective technical content sourced from engineers and other subject matter experts. Drive documentation quality standards and style guidelines. Produce documentation plans and content roadmaps.
AI-Powered Documentation Tools: Use your knowledge and experience with AI agents and LLMs to design and develop tools that convert subject matter expertise into well-designed, well-organized, easy-to-use technical documentation. Define the next-generation of Neuron's technical documentation as an AI-centered, interactive experience.
Process Optimization & Collaboration: Create plans to improve operational efficiency and productivity for the documentation process. Collaborate with Product Management to align documentation with product strategy.
About the team
Our team supports new members through knowledge-sharing and mentorship across diverse experience levels. We operate with startup-like velocity, prioritizing hands-on leadership and career growth through challenging projects that develop your expertise.
Diverse Experiences
AWS values diverse experiences. Even if you don't meet all qualifications, we encourage you to apply. Non-traditional career paths and alternative experiences are welcome.

Inclusive Team Culture:

At AWS, learning and curiosity drive us. Employee-led affinity groups and events like our Conversations on Race and Ethnicity (CORE) and Amaze Con conferences foster inclusion and celebrate our differences.
Work/Life Balance
We value work-life harmony and strive for flexibility, ensuring success at work doesn't require sacrifices at home.

Mentorship & Career Growth:

We provide endless knowledge-sharing, mentorship, and career-advancing resources to help you develop as a well-rounded professional.

About Amazon Annapurna Labs:

Amazon Annapurna Labs builds silicon and software innovations for AWS customers, including AWS Neuron, Inferentia and Trainium ML chips, AWS Nitro, and AWS Graviton.

About AWS

Amazon Web Services is the world's most comprehensive cloud platform, trusted by startups to Global 500 companies to power their businesses.

Basic Qualifications

• Bachelor's degree in Computer Science, English, Technical Writing, or a related field
• 5+ years experience as a technical writer, technical documentation manager, or product manager with technical docs experience and 3+ years (or equivalent) experience in Python and Git-based code tools development
• Strong knowledge and experience in documentation design and organization, information architecture, and web site development
• Knowledge of Sphinx and re Structured Text
• Excellent written and verbal communication abilities

Preferred Qualifications

• Current working knowledge of and experience with LLM-based tools and AI agents
• Experience with AI/ML infrastructure and model development
• Track record of collaborating with engineering teams
• Experience gathering and incorporating customer feedback
• Knowledge of Model Context Protocol and other structured AI agent support frameworks
• Basic project management skills (Jira, tracking, reporting)
  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 - 144,700.00 - 195,800.00 USD annually
USA, WA, Seattle - 125,800.00 - 170,200.00 USD annually