Senior Technical Writer, AWS Neuron

Senior Technical Writer, AWS Neuron

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 actively seeking a Senior Technical Writer to lead the documentation strategy and AI-based content contribution and automation initiatives for our AWS Neuron SDK. This SDK facilitates machine learning with our specialized AWS Inferentia and Trainium chips.

You will leverage your passion for technical writing and strategic thinking to design mechanisms and web experiences that integrate AI, empowering both Neuron customers and documentation contributors to create and utilize professional, consistent, and AI-ready technical documentation experiences. Operating at the intersection of machine learning, hardware acceleration, and developer experience, you will spearhead efforts to streamline documentation processes while upholding the high standards AWS customers expect from our documentation site and technical content.

Success in this role requires collaboration with multiple Neuron component engineering teams, Product Managers overseeing Neuron compiler, runtime, and NKI, and Solution Architects who support customers. You will develop profound knowledge of documentation design, organization, information architecture, AI agents, LLMs, and the Neuron SDK to effectively define documentation strategy and make informed content decisions.

The Ideal Candidate

The ideal candidate can effectively balance competing priorities across multiple workstreams and drive alignment among engineering and business stakeholders within a fast-moving, early-stage product environment. They possess excellent written and verbal communication abilities and have 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 serves as the software stack for executing deep learning and generative AI workloads on AWS Trainium and AWS Inferentia. It encompasses a compiler, runtime, training and inference libraries, and developer tools for monitoring, profiling, and debugging. Built upon an open-source foundation, Neuron natively supports PyTorch and JAX frameworks and popular ML libraries without requiring 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:

• Collaborate with multiple Neuron component engineering teams to plan content for Neuron releases.
• Drive the timely delivery of documentation materials.
• Develop code-based content that provides interactive experiences for Neuron's developer users.
• Manage and report on documentation efforts and progress within an Agile environment.

Documentation Strategy & Vision:

• Define and implement strategies for delivering effective technical content sourced from engineers and other subject matter experts.
• Drive the establishment and adherence to documentation quality standards and style guidelines.
• Produce comprehensive documentation plans and content roadmaps.

AI-Powered Documentation Tools:

• Utilize expertise in AI agents and LLMs to design and develop tools that transform subject matter expertise into well-designed, organized, and user-friendly technical documentation.
• Define the next-generation of Neuron's technical documentation as an AI-centered, interactive experience.

Process Optimization & Collaboration:

• Create plans to enhance operational efficiency and productivity for the documentation process.
• Collaborate closely with Product Management to ensure documentation aligns with overall product strategy.

Why AWS?

Our team fosters a supportive environment, aiding new members through knowledge-sharing and mentorship. We operate with startup-like agility, prioritizing hands-on leadership and career advancement via challenging projects that build expertise.

AWS values diverse experiences and encourages applications from individuals who may not meet every qualification. Non-traditional career paths and alternative experiences are welcome. Our inclusive culture is driven by learning and curiosity, supported by employee-led affinity groups and events that celebrate our differences. We also prioritize work-life harmony, striving for flexibility to ensure success at work does not compromise personal life. Additionally, we provide extensive knowledge-sharing, mentorship, and career-advancing resources to support your professional development.

About Amazon Annapurna Labs & AWS

Amazon Annapurna Labs develops silicon and software innovations for AWS customers, including AWS Neuron, Inferentia and Trainium ML chips, AWS Nitro, and AWS Graviton. 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
• 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

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)

Key skills/competency

• Technical Writing
• Documentation Strategy
• Information Architecture
• AI/LLM Tools
• Python Development
• Git Version Control
• AWS Neuron SDK
• Developer Experience
• Content Automation
• Agile Methodologies