Senior Technical Writer - AWS Neuron, Annapurna Labs

Amazon • Cupertino, CA

21h

About The Position

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

Requirements

• 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 reStructuredText
• Excellent written and verbal communication abilities
• Current working knowledge of and experience with LLM-based tools and AI agents
• Track record of collaborating with engineering teams
• Basic project management skills (Jira, tracking, reporting)

Nice To Haves

• Experience with AI/ML infrastructure and model development
• Experience gathering and incorporating customer feedback
• Knowledge of Model Context Protocol and other structured AI agent support frameworks

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.

Benefits

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

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Number of Employees: 5,001-10,000 employees