Senior Technical PM: ML Kernel & Compiler Interface

Amazon Web Services (AWS); Cupertino (CA); On-site; USD 173,900 - 235,200; Full time

Job summary

AWS Neuron is seeking a Technical Product Manager to define and drive the Neuron Kernel Interface (NKI) strategy, a compiler library enabling custom kernel development and optimization. You will lead NKI requirements from customer needs and drive developer experience for custom kernel development on AWS Neuron.

You will collaborate with engineering leadership, influence architecture decisions, and shape features that accelerate ML workloads on Trainium and Inferentia, while staying current with

Qualifications

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

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.

Skills

Technical product management

Roadmap strategy

Feature delivery

Stakeholder alignment

Education

Bachelor's degree

Job description

AWS Neuron is seeking a Technical Product Manager to define and drive the Neuron Kernel Interface (NKI) strategy, a compiler library enabling custom kernel development and optimization. You will lead NKI requirements from customer needs and drive developer experience for custom kernel development on AWS Neuron.

You will collaborate with engineering leadership, influence architecture decisions, and shape features that accelerate ML workloads on Trainium and Inferentia, while staying current with