Staff Product Manager, Training Infrastructure - Weights & Biases
Core
Own strategy and execution for infrastructure-native developer tools connecting experiment tracking with cluster health, hardware telemetry, and compute orchestration for frontier AI teams.
Role type
Staff Product Manager (Infrastructure & Developer Tools)
Builds
W&B Launch (job submission and compute orchestration), infrastructure insights, and new agentic/automated research workflows
Domain
AI Infrastructure / Deep Learning Platforms
Deliverable
production ML models | product features
Required skills
Product strategy, end-to-end ownership, technical fluency (Kubernetes, SLURM, job scheduling), cross-functional influence, clear communication, developer experience focus
Preferred skills
Deep learning frameworks (PyTorch, JAX), distributed training, LLM development workflows, experiment-tracking platforms, hyperparameter optimization
Technologies
Kubernetes, SLURM, W&B, CoreWeave platform
Responsibilities
Define and ship infrastructure-native products at the intersection of W&B and CoreWeave; Own W&B Launch evolution for the CoreWeave era; Partner with infrastructure and engineering teams to build integrations; Identify and resolve friction points in job submission and observability workflows; Explore new product concepts at the hardware-software boundary
Seniority
Staff, hands-on IC with strategic scope