Overview

Welcome to Baseten Training, a powerful product designed to streamline and manage the entire lifecycle of your machine learning model training.

​

Use Cases

Baseten Training provides a robust, scalable, and configurable platform for:

• Training machine learning models efficiently.
• Managing training artifacts, such as model checkpoints and logs.
• Seamlessly productionize trained models to scalable deployments.

​

Benefits of Using Baseten Training

Leveraging Baseten Training for your model training workflows offers several key advantages:

• Reproducibility: Ensure consistent training runs by precisely defining your environment, code, and configurations.
• Scalability: Easily scale your training jobs from single-node to multi-node distributed training to handle large datasets and complex models.
• Simplified Management: Organize, monitor, and manage your training projects and jobs in a centralized platform.
• Resource Flexibility: Configure compute resources (CPU, GPU, memory) tailored to the specific needs of each training job.
• Integrated Workflow: Transition from training to inference and evals seemlessly within the Baseten ecosystem.
• Artifact Management: Handle large artifacts like models, checkpoints, and datasets efficiently with Baseten storage.
• Framework Agnostic: Bring your favorite training framework or roll your own with Baseten’s framework-agnostic training API.

​

Get Started

Check out our Getting Started guide to get started with training on Baseten.

​

Go Deeper

Use the following resources to learn more about training on Baseten:

• CLI Reference
• SDK Reference
• API Reference