✦ Luna Orbit — AI & Machine Learning

Product Manager II - Model Lab

at Datadog

📍 New York, New York, USA Hybrid Posted March 11, 2026
Apply Now → ← Browse All Jobs

Position Details

Type Full-Time
Experience mid
Exp. Years 4+ years
Education Not specified
Category AI & Machine Learning

About this role

This role involves managing Datadog’s experiment tracking platform for AI and ML models, working with research and engineering teams to define product vision, and driving adoption of scalable ML training solutions.

Key Responsibilities

  • Define product vision for Model Lab
  • Lead discovery with AI teams
  • Design experiment tracking system
  • Partner with engineering
  • Drive customer adoption

Technical Overview

Focus on ML infrastructure, experiment tracking, distributed training, hyperparameter tuning, and reproducibility workflows, utilizing frameworks like PyTorch, TensorFlow, and JAX.

Ideal Candidate

The ideal candidate is a product manager with at least 4 years of experience in ML infrastructure, experiment tracking, and distributed training, with familiarity in frameworks like PyTorch, TensorFlow, or JAX. They are entrepreneurial, thrive in ambiguity, and can translate complex ML workflows into product features.

Must-Have Skills

Product ManagementExperience with ML infrastructureExperiment trackingDistributed trainingHyperparameter tuning

Nice-to-Have Skills

PyTorchTensorFlowJAXML training workflowsReproducibility workflows

Tools & Platforms

PyTorchTensorFlowJAXML frameworks

Required Skills

Product ManagementExperiment trackingML infrastructureData platformsAI/LLM systemsDistributed trainingHyperparameter tuningArtifact storageEvaluation pipelinesPyTorchTensorFlowJAX

Hard Skills

Product ManagementExperiment trackingML infrastructureData platformsAI/LLM systemsDistributed trainingHyperparameter tuningArtifact storageEvaluation pipelinesPyTorchTensorFlowJAX

Soft Skills

Entrepreneurial mindsetAmbiguity toleranceCross-functional collaborationTechnical translationPrioritization

Industry & Role

Industry SaaS
Job Function Product management for ML experiment tracking and infrastructure

Keywords for Your Resume

Product ManagementExperiment trackingML infrastructureData platformsAI/LLM systemsDistributed trainingHyperparameter tuningArtifact storageEvaluation pipelinesPyTorchTensorFlowJAXReproducibility workflows

Deal Breakers

Lack of experience in ML infrastructure or experiment tracking, No familiarity with ML frameworks, Inability to operate in ambiguous environments, No experience with distributed training, Lack of technical translation skills

Apply for this Position →

Get matched to jobs like this

Luna finds roles that fit your skills and career goals — no endless scrolling required.

Create a Free Profile