The Training API allows you to manage training projects, jobs, and related resources through a RESTful interface. This API is used to:

  • Monitor training job metrics and logs
  • Manage training jobs
  • Manage checkpoints and artifacts

Authentication

All Training API requests require authentication using an API key:

Authorization: Api-Key YOUR_API_KEY

Base URL

All Training API endpoints are relative to:

https://api.baseten.co/v1

Available Endpoints

Training Projects

MethodEndpointDescription
GET/training_projectsList all training projects
POST/training_projectsCreate a new training project

Training Jobs

Note: Creating Training Jobs via REST API is not supported at this time.

The following endpoints have the relative base path: /training_projects/{training_project_id}/jobs

MethodEndpointDescription
GET.../List all jobs in a project
GET.../{training_job_id}Get a specific training job
POST.../{training_job_id}/stopStop a training job
POST.../{training_job_id}/logsGet training job logs
POST.../{training_job_id}/metricsGet training job metrics
GET.../{training_job_id}/checkpointsList job checkpoints

Search endpoint:

MethodEndpointDescription
POST/training_jobs/searchSearch across all training jobs