Sr. PM-T, Kernels, AI/ML, Annapurna Labs
About the role
AWS Neuron is seeking a Technical Product Manager to lead the development of the Neuron Kernel Interface (NKI), a compiler library for custom kernel development and optimization. This role involves defining and executing product strategy, assessing technical implications, and driving developer experience for custom kernel development on AWS Neuron, which includes the Trainium and Inferentia chips.
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
About the team
The team is dedicated to supporting new members. It has a diverse mix of experience levels and tenures, fostering an environment that values knowledge sharing and mentorship. Senior members enjoy one-on-one mentoring and thorough, yet kind, code reviews. The team strives to assign projects that help its members develop their engineering expertise, empowering them to take on more complex tasks in the future.
Qualifications
- Bachelor's degree
- Experience owning/driving roadmap strategy and definition
- Experience with feature delivery and tradeoffs of a product
- Experience with technical product management
- Experience in project management methodologies, business analysis, or process improvement (preferred)
- Experience working directly with Engineers on product enhancements (preferred)