Senior Technical Writer - AWS Neuron, Annapurna Labs

Amazon • Cupertino, CA

1d

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.

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

Nice To Haves

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

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

• 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
• parental leave

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level