Principal Product Manager: AI/ML Training at Scale

Amazon; Cupertino (CA); On-site; USD 175,000 - 237,000; Full time

Job summary

Amazon in Cupertino, CA is seeking a Principal Technical Product Manager to define and drive product strategy for training software on Trainium. This role covers distributed training libraries, post-training workflows (RLHF, DPO, fine-tuning), reinforcement learning frameworks, and training performance optimization to scale frontier models from single-node experiments to thousands of nodes.

You will collaborate with engineering, applied scientists, and PMs for compiler, runtime, NKI, and

Qualifications

• 5+ years of technical product management with internet business experience
• 5+ years of working as a Technical Product Manager experience
• 3+ years of technical experience
• 7+ years of full product life cycle experience
• 5+ years of P&L management and pricing experience
• 5+ years of creating written docs for development of new products experience
• 5+ years of enterprise security product experience
• 5+ years of product management in the cloud computing technology space experience
• Bachelor's degree in computer science, engineering, math, finance, or economics
• Experience in taking a product from conception to market
• Experience delivering large-scale SaaS/PaaS/LaaS products

Responsibilities

• Define and execute training product strategy and roadmap
• Drive technical alignment across Neuron training libraries and infrastructure
• Partner with PMs responsible for compiler, NKI, runtime, and infrastructure
• Drive trade-offs between performance, scalability, and developer experience
• Define requirements for reusable training building blocks for end-to-end workflows
• Drive strategy for post-training workflows including RLHF, DPO, and fine-tuning
• Lead customer engagement to translate pain points into product requirements
• Define launch criteria and track adoption metrics

Skills

Technical product mgmt

Product lifecycle

Enterprise security

Cloud computing

P&L management

Written documentation

SaaS/PaaS/LaaS

Cross-functional leadership

Communication skills

Education

Bachelor's degree

Job description

Amazon in Cupertino, CA is seeking a Principal Technical Product Manager to define and drive product strategy for training software on Trainium. This role covers distributed training libraries, post-training workflows (RLHF, DPO, fine-tuning), reinforcement learning frameworks, and training performance optimization to scale frontier models from single-node experiments to thousands of nodes.

You will collaborate with engineering, applied scientists, and PMs for compiler, runtime, NKI, and