Principal Pmt-es - Ai/ml Training, Annapurna Labs

Amazon

United States
**
Aws trainium and neuron
Distributed training libraries
Post-training workflows
** Amazon is seeking a Principal Technical Product Manager for AWS Neuron to define and drive the product strategy for training software on Trainium, focusing on large-scale AI/ML model training and optimization. The role involves collaborating with various stakeholders to improve training workflows and customer engagement while fostering a diverse and inclusive team culture. **

Job Summary

  • Define and drive product strategy for training software on Trainium, enabling customers to train frontier models at scale.
  • Work with customers to identify key improvements and define how Neuron supports the AI/ML ecosystem and training workflows.
  • Partner with engineering, applied science, and PMs to define product strategy and make informed technical decisions for training capabilities.

Matching Summary

Match Score: 75

** Amazon is seeking a Principal Technical Product Manager for AWS Neuron to define and drive the product strategy for training software on Trainium, focusing on large-scale AI/ML model training and optimization. The role involves collaborating with various stakeholders to improve training workflows and customer engagement while fostering a diverse and inclusive team culture. **

Skills & Requirements

Must-have

  • AWS Trainium and Neuron
  • Distributed training libraries
  • Post-training workflows
  • Reinforcement learning frameworks
  • Training performance optimization
  • Large-scale model training
  • Customer engagement and enablement

Nice-to-have

  • Navigating ambiguity
  • Balancing competing priorities
  • Driving alignment across stakeholders
  • Startup-like velocity
  • Knowledge sharing and mentorship

Key Requirements

  • Solid understanding of large-scale model training
  • Experience with distributed training architectures
  • Experience with post-training workflows
  • Experience with reinforcement learning
  • Ability to assess technical implications
  • Excellent written and verbal communication

Work Rights

Not specified

Tailored Resume

Cover Letter