Principal AI Product Lead - Agentic Workflows

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

Job summary

Annapurna Labs (U.S.) Inc. in Cupertino seeks a Principal Technical Product Manager to lead the strategy for agentic AI systems that accelerate Trainium adoption.

You will own end-to-end product experiences for autonomous workflows and generative AI porting, optimization, and tuning of ML models for Neuron, enabling customers to migrate from other platforms with minimal manual intervention. You will shape customer interactions across IDE-based experiences to production pipelines, collaborate

Qualifications

• 7+ years of technical product management experience in AI/ML platforms or automation systems.
• Experience with AI/ML workflows, agentic AI architectures, or multi-agent systems.
• Demonstrated ability to define metrics and success criteria for AI/ML products.

Responsibilities

• Drive product strategy and roadmap for agentic workflows that automate model porting and performance optimization.
• Own the product experience for agentic workflows from interactive transformations to fully autonomous pipelines.
• Define launch plans, documentation, and customer enablement for Neuron SDK and developer tools.

Skills

Technical product management

LLM based applications

Agent architectures

Metrics driven development

Communication skills

AI/ML platforms

Developer tools

IDE integrations

Education

Bachelor's degree in CS/engineering/math

Tools

PyTorch

JAX

Hugging Face

Job description

Annapurna Labs (U.S.) Inc. in Cupertino seeks a Principal Technical Product Manager to lead the strategy for agentic AI systems that accelerate Trainium adoption.

You will own end-to-end product experiences for autonomous workflows and generative AI porting, optimization, and tuning of ML models for Neuron, enabling customers to migrate from other platforms with minimal manual intervention. You will shape customer interactions across IDE-based experiences to production pipelines, collaborate