Principal Product Manager - Kernels, AI/ML, Annapurna Labs jobs in United States
cer-icon
Apply on Employer Site
company-logo

Amazon Web Services (AWS) · 12 hours ago

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

Amazon Web Services (AWS) is seeking an experienced Technical Product Manager to define and drive product strategy for the Neuron Kernel Interface, enhancing machine learning acceleration software. The role involves collaborating with engineering teams to assess technical implications, drive product features, and improve developer experience for custom kernel development on AWS Neuron.

Agentic AIConsultingDevOpsInformation TechnologySoftwareWeb Development
check
H1B Sponsor Likelynote

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

Product strategyCompiler technologiesMachine learning systemsTechnical product managementFeature deliveryProject managementCollaborationCommunicationProblem-solving

Required

Bachelor's degree
Experience owning/driving roadmap strategy and definition
Experience with feature delivery and tradeoffs of a product
Experience technical product management
Proven track record of driving product strategy and owning roadmap definition in complex technical environments
Experience delivering features with deep understanding of technical trade-offs and product implications
Strong ability to contribute to and influence engineering discussions around technology decisions and strategy
Demonstrated success in representing customer needs and driving alignment during executive-level prioritization
Track record of delivering results in fast-paced, ambiguous environments, particularly in early-stage programs
Experience with modern software development and collaborative open-source projects
Solid understanding of compiler technologies, computer architecture, and ML systems
Ability to assess technical implications of compiler architecture decisions
Understanding of customer needs and ability to drive developer experience strategy
Experience with ML frameworks, model architectures, and distributed computing

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 Web Services (AWS)

company-logo
Launched in 2006, Amazon Web Services (AWS) began exposing key infrastructure services to businesses in the form of web services -- now widely known as cloud computing.

H1B Sponsorship

Amazon Web Services (AWS) 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
Late Stage
Total Funding
unknown
Key Investors
BIRD Foundation
2025-01-22Grant

Leadership Team

leader-logo
Matt Garman
Chief Executive Officer
linkedin
leader-logo
Anand Desikan
CTO, CXO Advisor, and Enterprise Technologist
linkedin
Company data provided by crunchbase