Principal PM: Frontier Model Training on Trainium

Socket.dev; Seattle (WA); On-site; USD 152,000 - 206,000; Full time

Job summary

Amazon Annapurna Labs in Seattle is hiring a Principal Technical Product Manager to define and drive the training software strategy for Trainium. You will shape distributed training libraries, post-training workflows, and performance optimization to empower researchers and operators to train frontier models at scale.

You will partner with engineering, applied science, and PM teams to deliver reusable building blocks, drive adoption, and lead end-to-end launches across the training ecosystem on

Qualifications

• 5+ years in technical product management with internet business experience.
• 5+ years as a Technical Product Manager in software.

Responsibilities

• Define and execute training product strategy and roadmap for Trainium.
• Lead post-training workflows including RLHF, DPO, and fine-tuning at scale.

Skills

Technical product management

Internet business experience

Cross-functional collaboration

Strategic thinking

Executive communication

Customer engagement

GTM planning

Distributed training basics

RLHF familiarity

Product lifecycle

Education

Bachelor's degree in CS/Engineering/Math/Finance/Economics

Job description

Amazon Annapurna Labs in Seattle is hiring a Principal Technical Product Manager to define and drive the training software strategy for Trainium. You will shape distributed training libraries, post-training workflows, and performance optimization to empower researchers and operators to train frontier models at scale.

You will partner with engineering, applied science, and PM teams to deliver reusable building blocks, drive adoption, and lead end-to-end launches across the training ecosystem on