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
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
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
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 CompanyTotal Funding
$8.11BKey Investors
AmazonKleiner Perkins
2023-01-03Post Ipo Debt· $8B
2001-07-24Post Ipo Equity· $100M
1997-05-15IPO
Recent News
2026-02-09
The Motley Fool
2026-02-09
Company data provided by crunchbase