veri-sdk

Python SDK for Veri — RL post-training platform.

Install

uv pip install veri-sdk
# or
pip install veri-sdk

Quickstart

from veri_sdk import Client

client = Client(api_key="your-api-key", base_url="https://api.veri.studio")

# Upload a dataset
dataset = client.datasets.upload("training_data.jsonl", name="my-dataset")

# Start a GRPO training job — the reward Python source is attached to the
# job inline (rewards are not stored as a resource)
job = client.training_jobs.create(
    base_model="Qwen/Qwen3-4B",
    dataset_id=dataset.id,
    reward_source=open("reward.py").read(),
    output_name="my-fine-tuned-model",
    hyperparameters={
        "learning_rate": 1e-6,
        "max_steps": 100,
        "rollouts_per_prompt": 4,
        "max_response_length": 512,
    },
)

print(f"Job {job.id} — status: {job.status}")

# Wait for completion
job.wait(poll_interval=15)
print(f"Done! Status: {job.status}")

# Download checkpoint
if job.download_url:
    job.download("./checkpoints")

Data Sources

# Upload JSONL file
dataset = client.datasets.upload("data.jsonl")

# Connect to S3
dataset = client.datasets.connect(
    name="my-s3-data",
    source_type="s3",
    source_uri="s3://my-bucket/data.jsonl",
    credentials={"aws_access_key_id": "...", "aws_secret_access_key": "..."},
)

# Connect to HuggingFace
dataset = client.datasets.connect(
    name="gsm8k",
    source_type="hf",
    hf_dataset="gsm8k",
    hf_config={"split": "train", "column_mapping": {"question": "prompt"}},
)

# Connect to a database
dataset = client.datasets.connect(
    name="prod-prompts",
    source_type="postgres",
    db_connection="postgres://user:pass@host/db",
    db_query="SELECT prompt, answer FROM training_data",
)

# Validate before connecting
result = client.datasets.validate(
    source_type="hf", hf_dataset="gsm8k", hf_config={"split": "train"}
)
print(f"Valid: {result['valid']}, Rows: {result['num_rows']}")

GPU Selection

# Specify the GPU config explicitly.
job = client.training_jobs.create(
    base_model="Qwen/Qwen3-4B",
    gpu_type="A100-80GB",
    gpu_count=2,
    ...
)

Checkpoint Destination

# Default: Veri-managed storage
job = client.training_jobs.create(...)

# Your own S3 bucket
job = client.training_jobs.create(
    ...,
    checkpoint_destination={
        "type": "s3",
        "uri": "s3://my-bucket/checkpoints/",
    },
)

List & Manage

# List your datasets
datasets = client.datasets.list()

# List jobs by status
running_jobs = client.training_jobs.list(status="running")

# Cancel a job
client.training_jobs.cancel(job.id)

Metadata

Release files for veri-sdk 0.2.41

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for veri-sdk 0.2.41
File	Size	Uploaded
veri_sdk-0.2.41.tar.gz	298.2 kB	Oct 3, 2026	Details

Built distribution (wheel)

Table of built distributions (wheels) for veri-sdk 0.2.41
File	Interpreter	ABI	Platform	Reset
veri_sdk-0.2.41-py3-none-any.whl 161.7 kB Oct 3, 2026	Python 3	none	any	Details

Total release size: 459.9 kB

Release files / veri_sdk-0.2.41.tar.gz

Download URL	veri_sdk-0.2.41.tar.gz
Size	298.2 kB
Tags	Source
SHA-256 checksum How to use checksums	`5548ca6e509df56d27be65c58f307cb115e4474b78ac8f29108b9252d50fc1a8`
BLAKE2b-256 checksum How to use checksums	`098593457f31dc717f8d3d36a486c93640cd976f6aa5fa22c647fd5b134b9c2e`
Upload date	Oct 3, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`uv/0.11.2 {"installer":{"name":"uv","version":"0.11.2","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}`

Release files / veri_sdk-0.2.41-py3-none-any.whl

Download URL	veri_sdk-0.2.41-py3-none-any.whl
Size	161.7 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`600b8b8cc0b5a674eb3e57577553d9651bd2943e08145c69bc2c209dc84d4893`
BLAKE2b-256 checksum How to use checksums	`4c11db8b5ba18c0d506e17e43e49b6b246adbab46453633c0a15a1d1d1d43477`
Upload date	Oct 3, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`uv/0.11.2 {"installer":{"name":"uv","version":"0.11.2","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}`