Principal Product Manager: AI/ML Training at Scale

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