Principal Technical Program Manager , Annapurna ML
About the role
Annapurna ML Neuron delivers the software that powers existing and upcoming Trainium families of Machine Learning Accelerated EC2 instances, building a compiler, collective communication, drivers, runtime, and a fully integrated PyTorch and JAX stack for distributed training and inference. The Principal Program Manager will own the software technology bring‑up lifecycle from early hardware design ideation through lab and data‑center deployment, develop, debug, and qualify full servers for customers and Neuron developers, secure and allocate prototype and production Trainium instances, establish dashboards and utilization metrics, identify new use‑cases and gaps, scope and deliver large end‑to‑end projects, collect business and systems requirements, write specifications, drive schedules, manage production launch, lead cross‑functional design/implementation between Neuron, Trainium hardware, internal and external teams, make trade‑offs to optimize time‑to‑production, communicate goals and outcomes, anticipate bottlenecks, provide escalation management, break complex projects into manageable pieces, develop functional specifications, and deliver them timely.
Responsibilities
- Drive execution of projects
- Provide technical direction with limited assistance
- Lead cross‑functional project meetings with VP‑level audience and stakeholders
- Lead milestone reviews
- Present project status to the executive team
- Work with Engineering leadership, Product Management, and Business Development to help define requirements and roadmaps
- Work with Engineering leadership to drive efficiencies in engineering development and ensure the right things are delivered to customers at the right time
- Collect business and systems requirements from internal and external customers
- Write specifications
- Drive project schedules from design to release
- Manage the production launch
- Lead and coordinate design/implementation efforts between Neuron, Trainium hardware teams, internal teams, and external customers
- Make appropriate trade‑offs to optimize time‑to‑production
- Clearly communicate goals, roles, responsibilities, and desired outcomes to internal