The mixtrain package provides the Python SDK and CLI for the mixtrain.ai platform.

Installation

Install mixtrain SDK

Using uv

uv add mixtrain

or if you use pip

pip install mixtrain

Install mixtrain CLI

To install the mixtrain CLI globally:

curl -LsSf https://mixtrain.ai/install | sh

This uses uv tool install mixtrain under the hood and installs uv first if needed.

If you already use uv:

uv tool install mixtrain

Or if you use pipx:

pipx install mixtrain

Refer to mixtrain --help for the full list of commands.

Quick Start

After installation, authenticate with your API key:

mixtrain login

Train a model on your dataset with a workflow:

from mixtrain import Workflow

workflow = Workflow("smolvlm-finetune")
result = workflow.run(
    dataset="my-dataset",
    base_model="smolvlm-2",
    steps=10000,
)

or run an external model:

model = Model("nano-banana-pro-edit")
result = model.run({"prompt": "Make it sunset", "image_url": "https://example.com/photo.jpg"})
print(result.image.url)

Next Steps

• Authentication - Configure API keys and credentials
• Build with Claude Code - Use AI to build models and workflows
• Examples - See complete working examples
• Python SDK Reference - Complete API reference
• CLI Reference - Command-line tools