MixtrainDocsBlog

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 mixtrain CLI globally, using 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:

from mixtrain import Model

model = Model("smolvlm-2")
model.train(dataset="my-dataset", 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

On this page