Apply on Employer Site

Amazon · 4 hours ago

Principal Product Manager - Kernels, AI/ML, Annapurna Labs

Cupertino, California, USA

Full-time

Onsite

Lead/Staff

$198K/yr - $268K/yr

Amazon Annapurna Labs is seeking an experienced Technical Product Manager to define and drive product strategy for the Neuron Kernel Interface, a compiler library for custom kernel development in machine learning. The role involves collaborating with engineering teams to assess technical implications, drive product alignment, and enhance developer experience for AWS Neuron's machine learning acceleration software.

Artificial Intelligence (AI)DeliveryE-CommerceFoundational AIRetail

H1B Sponsor Likely

Responsibilities

Drive and execute product strategy and roadmap working backwards from customer requirements in collaboration with engineering technical leadership

Assess technical implications of product architecture and optimization decisions

Drive technical alignment across Neuron components, Neuron workflows and dependencies

Work directly with software engineering teams to define and execute on new features

Produce clear and concise documents such as PRFAQ and PRD documents

Write user stories, and validate features meet developer needs

Drive feature discussions with customers, engineering, and other stakeholders

Anticipate bottlenecks, manage risk and escalations, balancing technical constraints

Find opportunities to innovate on behalf of our customers, design features related to these opportunities, and always push to improve our product developer experience

Build ecosystem partnerships and stay connected with industry trends

Represent the product in relevant industry events

Qualification

Technical product managementRoadmap strategyCompiler technologiesMachine learning systemsBusiness analysisProcess improvementProject management methodologiesInfluencing engineering discussionsCollaborationCommunication

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

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

Glassdoor3.7

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

Founded in 1994

Seattle, Washington, USA

10001+ employees

https://amazon.com

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

Douglas J. Herrington

CEO, Worldwide Amazon Stores

Werner Vogels

VP & CTO

Recent News

TechRadar.com

Amazon is pushing Alexa+ upgrades on Prime members – and some users aren't happy about it

2026-01-16

Japan Today

AI pendants back in vogue at tech show after early setback

2026-01-16

Chicago Sun-Times - Business

State high court hears arguments in Amazon overtime dispute

2026-01-16

Company data provided by crunchbase