Principal PM, Agentic AI for ML Acceleration

Socket.dev

Cupertino (CA)

On-site

USD 208,000 - 282,000

Full time

7 days ago
Be an early applicant
Application generator

Turn this role into an interview — a resume and cover letter built around what this employer wants.

Get past ATS filters

Job summary

Amazon's AWS Neuron team seeks a Principal Technical Product Manager to lead the strategy for agentic AI systems that accelerate Trainium adoption. You will own end-to-end product experiences for autonomous and interactive workflows, spanning interactions to production pipelines, and collaborate with engineering, data science, and field teams to translate customer needs into scalable capabilities.

The role requires deep knowledge of LLM-based automation, model optimization, and developer

Qualifications

  • 7+ years of experience in technical product management
  • Experience with AI/ML platforms, developer tools, or automation systems
  • Demonstrated experience with LLM based applications, including prompt engineering or agent architectures
  • Experience defining metrics and success criteria for AI/ML products
  • Excellent written and verbal communication abilities

Responsibilities

  • Drive product strategy and roadmap for agentic workflows that automate model porting and performance optimization.
  • Own the product experience for agentic workflows, from interactive transformations to fully autonomous pipelines.
  • Engage with customers across the lifecycle to support technical evaluation and post-adoption tuning.
  • Own end-to-end launch strategy for agentic capabilities with cross-functional coordination.
  • Drive strategy for knowledge systems to learn from past successes and failures.
  • Drive technical alignment across agentic infrastructure, autonomous workflows, and optimization systems.

Skills

Technical product management
LLM based applications
Metrics driven development
Developer tools experience

Education

Bachelor's degree (CS/Engineering/Math)

Tools

PyTorch
JAX
Hugging Face

Job description

Amazon's AWS Neuron team seeks a Principal Technical Product Manager to lead the strategy for agentic AI systems that accelerate Trainium adoption. You will own end-to-end product experiences for autonomous and interactive workflows, spanning interactions to production pipelines, and collaborate with engineering, data science, and field teams to translate customer needs into scalable capabilities.

The role requires deep knowledge of LLM-based automation, model optimization, and developer

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Principal Product Manager — Agentic AI Platforms
Principal Product Manager — Agentic AI Platforms

Amazon • Cupertino (CA)

On-site
USD 208,000 - 282,000
Principal Product Manager: Agentic AI & Pipelines
Principal Product Manager: Agentic AI & Pipelines

Amazon • Cupertino (CA)

On-site
USD 208,000 - 282,000
Stock options
Health benefits
Principal Product Manager — AI/ML Training on Trainium
Principal Product Manager — AI/ML Training on Trainium

Amazon • Seattle (WA)

On-site
USD 152,000 - 206,000
Health insurance
RSUs
Paid time off
+1
Principal PM, AI/ML Training at Scale
Principal PM, AI/ML Training at Scale

Annapurna Labs (U.S.) Inc. • Cupertino (CA)

On-site
USD 180,000 - 260,000
Sr. Product Manager - AI/ML Training, Annapurna Labs
Sr. Product Manager - AI/ML Training, Annapurna Labs

Annapurna Labs (U.S.) Inc. • Cupertino (CA)

On-site
USD 180,000 - 260,000
Senior AI SDK Engineering Manager (Neuron)
Senior AI SDK Engineering Manager (Neuron)

Amazon Web Services (AWS) • Seattle (WA)

On-site
USD 220,000 - 298,000
Principal Product Manager - Agentic AI Platforms
Principal Product Manager - Agentic AI Platforms

Amazon Web Services (AWS) • Seattle (WA)

On-site
USD 181,000 - 245,000
Principal AI Training Platform Lead
Principal AI Training Platform Lead

Amazon • Cupertino (CA)

On-site
USD 208,000 - 282,000
Senior AI SDK Lead for Neuron on Trainium/Inferentia
Senior AI SDK Lead for Neuron on Trainium/Inferentia

Relha LLC • Seattle (WA), Northern (KY)

Hybrid
USD 220,000 - 298,000
Applied Scientist, ML Systems for AWS Neuron
Applied Scientist, ML Systems for AWS Neuron

Amazon • Cupertino (CA)

On-site
USD 171,600 - 222,200
RSUs
Health insurance
401(k) matching
+1