Principal Product Manager — Agentic AI Platforms

Amazon; Cupertino (CA); On-site; USD 208,000 - 282,000; Full time

Job summary

Amazon Annapurna Labs in Cupertino seeks a Principal Product Manager to drive the strategy and end-to-end experience for agentic workflows that accelerate adoption of AWS Trainium and Neuron. You will own continuous product improvements spanning interactive and autonomous pipelines, integrating with the Neuron SDK and developer tools.

The ideal candidate combines technical PM experience in AI/ML platforms with leadership across multiple workstreams, delivering measurable impact and customer

Qualifications

• Experience with agentic AI systems and multi-agent architectures.
• Knowledge of ML model training, tuning, deployment workflows.
• Familiarity with developer tools, IDE integrations, and AI tooling.

Responsibilities

• Drive product strategy for agentic workflows and automated model porting.
• Own end-to-end product experience for autonomous pipelines.
• Define requirements for reusable building blocks and end-to-end automation.
• Coordinate cross-functional launch activities and customer communications.
• Define success metrics and adoption strategies.

Skills

Agentic AI systems

ML model architectures

Model optimization

Data platforms knowledge

Developer tools integration

Reinforcement learning

Compiler/kernel knowledge

Open source collaboration

Tools

PyTorch

JAX

Hugging Face

Job description

Amazon Annapurna Labs in Cupertino seeks a Principal Product Manager to drive the strategy and end-to-end experience for agentic workflows that accelerate adoption of AWS Trainium and Neuron. You will own continuous product improvements spanning interactive and autonomous pipelines, integrating with the Neuron SDK and developer tools.

The ideal candidate combines technical PM experience in AI/ML platforms with leadership across multiple workstreams, delivering measurable impact and customer