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

About the role

AWS Neuron is seeking a Technical Product Manager to lead the development and optimization of the Neuron Kernel Interface (NKI), a compiler library enabling custom kernel development for AWS Machine Learning chips. This role involves defining and executing product strategy, collaborating with engineering, and driving customer needs.

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

Requirements

• 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

Qualifications

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

Preferred Qualifications

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

About AWS Neuron

AWS Neuron is the software stack for Trainium and Inferentia, AWS Machine Learning chips. It delivers best-in-class ML performance in the cloud. The role involves leading NKI requirements, defining kernel library features, and enhancing developer experience for custom kernel development.

About the team

The team supports new members and fosters a culture of inclusion. Senior members enjoy one-on-one mentoring and thorough code reviews.

About Amazon Annapurna Labs

Annapurna Labs focuses on innovation in silicon and software for AWS customers. The team covers multiple disciplines including silicon engineering, hardware design, software, and operations.

About AWS Utility Computing (UC)

As a member of the UC organization, you'll support the development and management of Compute, Database, Storage, Platform, and Productivity Apps services in AWS, including support for specialized security solutions.

About AWS

AWS is the world's most comprehensive and broadly adopted cloud platform. It offers a suite of products and services to power businesses globally, including comprehensive benefits such as health insurance, retirement plans, and parental leave.