B

Baseten

Runs AI models for production software

Product Manager - Inference Platform

Technical Product ManagementPosted on Nov 11
Full-timeHybrid

About the Job

Requirements

  • Experience building and shipping AI or machine-learning products or infrastructure, particularly where performance, scalability, or production behavior mattered.
  • A technical background in software engineering, machine-learning engineering, research, or equivalent hands-on experience.
  • Strong understanding of model-serving architectures, GPU economics, and latency, throughput, quality, and cost tradeoffs.
  • Experience working closely with researchers or infrastructure engineers to bring technically complex capabilities into production.
  • A rigorous approach to benchmarking, experimentation, and performance measurement.
  • Strong product judgment and the ability to translate deep technical work into clear customer value.

Responsibilities

  • Work with inference engineering and research teams to bring new performance capabilities to customers.
  • Translate advances in batching, caching, quantization, speculative decoding, and serving architectures into useful product features.
  • Define benchmarks and evaluations that demonstrate meaningful improvements in latency, throughput, quality, and cost.
  • Understand customer workloads and help prioritize the optimizations that will have the greatest impact.
  • Make performance features easier to configure, evaluate, and adopt in production.
  • Help customers understand tradeoffs across hardware, model configurations, autoscaling, and routing.
  • Partner with engineering on release readiness, compatibility, and regression testing so customers can trust new capabilities.

Desired Qualifications

  • Hands-on experience with PyTorch, vLLM, TensorRT-LLM, SGLang, NVIDIA Dynamo, or similar technologies.
  • Experience with open-source inference projects, model optimization, or ML systems research.

About the Company

Baseten

Runs AI models for production software

Baseten runs AI models for software companies that need fast, reliable responses inside their products, from text generation to speech and image processing. Its Inference Stack combines optimized model runtimes with infrastructure that deploys, serves, and monitors custom or open models in production. Teams package models with its open source Truss tool, while Model APIs let developers test ready-to-use models and Training runs jobs that adapt models to their own data. Customers pay for computing time or tokens processed, with negotiated plans for larger deployments and support needs. Baseten was founded in 2019 and is headquartered in San Francisco.

Company size
201-500
Company stage
Series F
Total funding
$2.1B
Headquarters
San Francisco, CA
Founded
2019

Recently Posted Jobs

Explore more opportunities at Baseten

See more jobs