Principal. Tech Product Manager - Neuron Runtime and Cloud ML Infrastructure, Annapurna Labs jobs in United States
cer-icon
Apply on Employer Site
company-logo

Amazon · 8 hours ago

Principal. Tech Product Manager - Neuron Runtime and Cloud ML Infrastructure, Annapurna Labs

Amazon is a leading cloud platform provider, and they are seeking a Principal Tech Product Manager to drive the developer experience for running high-performance ML workloads on AWS Trainium. The role involves owning product strategy, understanding customer needs, and collaborating with engineering teams to ensure optimal performance and scalability.

RetailArtificial Intelligence (AI)E-CommerceLogisticsDeliveryFoundational AI
check
H1B Sponsor Likelynote

Responsibilities

Own product strategy and roadmap
Guide trade-offs between performance, scalability, and developer experience
Write PRFAQs and PRDs
Understand deployment challenges, orchestration needs, and infrastructure pain points
Represent customer needs in executive prioritization
Drive alignment across Neuron components (Runtime, Kernel Driver, Collective Communication, container infrastructure) and AWS services
Partner with training, inference, and performance PMs
Write user stories and define success metrics
Enable customers (Anthropic, Databricks, AWS teams) to deploy, monitor, and operate ML workloads at scale through container orchestration, resource management, health monitoring, and observability

Qualification

Technical product managementAWS cloud servicesResource management systemsML monitoringObservabilityDeveloper toolsDistributed systemsCommunicationProject management methodologies

Required

Bachelor's degree
Experience owning/driving roadmap strategy and definition
Experience with feature delivery and tradeoffs of a product
Experience technical product management

Preferred

Experience working directly with Engineers on product enhancements
Experience in project management methodologies, business analysis, or process improvement

Benefits

Comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage)
401(k) matching
Paid time off
Parental leave

Company

Amazon is a tech firm with a focus on e-commerce, cloud computing, digital streaming, and artificial intelligence.

H1B Sponsorship

Amazon has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (22803)
2024 (21175)
2023 (19057)
2022 (24088)
2021 (12233)
2020 (14881)

Funding

Current Stage
Public Company
Total Funding
$8.11B
Key Investors
AmazonKleiner Perkins
2023-01-03Post Ipo Debt· $8B
2001-07-24Post Ipo Equity· $100M
1997-05-15IPO

Leadership Team

leader-logo
Douglas J. Herrington
CEO, Worldwide Amazon Stores
linkedin
leader-logo
Werner Vogels
VP & CTO
linkedin
Company data provided by crunchbase