Principal Product Manager, AI Agent Infrastructure
Not sure if you're a good fit?
Upload your resume and TixelJobs AI will compare it against Principal Product Manager, AI Agent Infrastructure at OKX. Get a match score, missing keywords, and improvement tips before you apply.
Free preview · Your resume stays private
About the Role
Who We Are
About The Opportunity
We are looking for a Product Manager who can drive the foundational infrastructure behind AI agent products. As agentic AI rapidly moves from research to production, the platforms that power agent development, evaluation, and deployment have become mission-critical. You will own the product vision, strategy, and execution for these infrastructure systems — ensuring our teams can build, test, and ship high-quality AI agents at scale.
This role is ideal for someone who has worked on AI/ML platforms, developer tools, or data infrastructure, and wants to apply that experience to the frontier of agentic AI.
What We Are Looking For
You have experience building the platforms and tools that empower teams to develop, evaluate, and operate AI agents at production quality. Ideal candidates will have background in one or more of the following areas:
- Data Management & Labeling — annotation platforms, training data pipelines, dataset versioning, data quality assurance, and data governance
- Agent Evaluation & Quality Assurance — benchmarking, automated evaluation, A/B testing, human-in-the-loop review, and feedback-driven model improvement
- Workflow Orchestration & Agent Runtime — multi-step task orchestration, agent observability, debugging tooling, and developer-facing configuration platforms
- Tool / Plugin Ecosystem Management — tool registries, API connector platforms, developer marketplaces, permission management, and security sandboxing
What You’ll Be Doing
- Own the end-to-end product lifecycle for AI agent infrastructure: from problem discovery and requirement definition, through development and launch, to post-launch iteration
- Define product vision, strategy, and roadmap for infrastructure that serves internal AI product teams as primary customers
- Partner closely with engineering, AI/ML research, and business stakeholders to translate complex technical capabilities into well-scoped, shippable product requirements
- Establish evaluation frameworks and quality metrics that drive continuous improvement of agent performance, reliability, and safety
- Build feedback loops connecting production signals (user ratings, task success/failure, escalations) back into model and data improvement pipelines
- Define platform SLAs, KPIs, and capacity plans; balance scalability and reliability with fast iteration
- Stay current with the rapidly evolving AI agent landscape and translate emerging capabilities (new models, tool-use paradigms, orchestration patterns) into infrastructure opportunities
What We Look For In You
- 5+ years of product management experience, with at least 2 years focused on AI/ML products or developer infrastructure
- Demonstrated experience shipping AI products or platforms to production (consumer or enterprise)
- Strong technical acumen: comfortable discussing model architectures, prompt engineering, API design, and system trade-offs with engineers
- Solid understanding of the AI agent development lifecycle: from data collection and annotation, through model training and evaluation, to production deployment and monitoring
- Data-driven mindset with experience defining KPIs, running experiments, and making decisions under uncertainty
- Strong written and verbal communication in English; Mandarin Chinese is a significant plus
Nice-To-Haves
- Hands-on experience with AI agent frameworks or platforms (e.g., Claude Skills, Anthro
Ready to apply?
This job is active. Apply now to get in early.