Flash — managed LoRA post-training (SFT/GRPO) for verifiers environments, driven by the `flash` CLI
Project description
Flash
Managed LoRA post-training service: SFT and GRPO on managed GPUs across multiple providers — RunPod Flash (serverless queue; RTX 4090/5090 classes) and Vast.ai (rented verified-datacenter instances; L40S / RTX Pro 4000 / A100 classes). The allocator picks the cheapest GPU class that fits the run across both providers.
Scope
flash train <cfg.toml>/ control-planePOST /runs— submit a training job; one dedicated GPU per run, supervised server-side (stall watchdog, bounded auto-retry resuming from the last streamed checkpoint, endpoint GC).flash deploy(scale-to-zero or always-on),flash chat— serving for trained adapters.- Verifiers-only environments. Every run names a Prime Intellect
verifiersenvironment by its published Hub slug ([environment] id = "owner/name"). Scaffold a local env, publish it withflash env push, then reference it by id. The worker wraps it viaflash/envs/adapter.py. There are no built-in task environments and no freesolo bridge. Single-turn environments are fully supported (SFT/GRPO/eval).
Layout
flash/catalog.py— curated model catalog (Qwen3 dense supported tier; Qwen3.5/3.6 experimental tier) +model_policy = "allow"VRAM-fit check + each model'sthinkingcapability (opt-in reasoning modethinking = true)flash/schema.py,flash/spec.py— TOML →JobSpecflash/runner.py— server-side run supervisor (durable job handle, retries, cost guard, endpoint GC)flash/providers/— RunPod Flash + Vast.ai provider subtrees (pricing, gpus, durable submit/poll, preflight) behind onebase.Providerprotocol, with a cross-providerallocator.pythat picks the cheapest fitting classflash/engine/— the on-GPU worker (TRL + colocated vLLM rollouts) and the shared recipe; SFT targets and RL rewards route through the active environment (task-specific grading lives with its example, not in the engine)flash/envs/— environment machinery: registry and theadapterthat wraps Prime Intellect / Hubverifiersenvironments onto the worker's interfaceflash lab setup/flash env init— scaffold a starter local verifiers env and a ready-to-run config to start fromflash/serve/,flash/server/— adapter serving and the FastAPI control plane (run operator-side via the separateflash-servercommand)flash/mcp/— stdio MCP bridge for coding agentsDockerfile— the control-plane image (used by the repo docker-compose)tests/— pytest suite (CPU-only; offline-by-default, no GPU/network)
Local commands
cd flash
uv sync --extra server
uv run pytest # CPU tests (offline-by-default, no GPU/network)
uv run ruff check . && uv run ruff format .
uv run flash --help
uv run flash-server # control plane (operator-side, run once)
The control plane owns provider credentials: RUNPOD_API_KEY is always required
(RunPod is the default substrate), VAST_API_KEY is opt-in (only checked when set),
plus the shared HF_TOKEN.
The artifact repo is per-run (the run TOML's [train] hf_repo), not an
operator-wide env var. Clients authenticate with their freesolo API key (flash login).
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file freesolo_flash-0.2.7.tar.gz.
File metadata
- Download URL: freesolo_flash-0.2.7.tar.gz
- Upload date:
- Size: 570.8 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.11.14 {"installer":{"name":"uv","version":"0.11.14","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
ee7b894b487bc519e24bfbbc4fc98dcdf5d4302824bb649fd6d8c25aee7db885
|
|
| MD5 |
f24bac4361d6c459082b661facf923c9
|
|
| BLAKE2b-256 |
90ceb3398b108c19c2e4e0624280728a9d06afda84395d272fdb2a8dae5e6aed
|
File details
Details for the file freesolo_flash-0.2.7-py3-none-any.whl.
File metadata
- Download URL: freesolo_flash-0.2.7-py3-none-any.whl
- Upload date:
- Size: 262.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.11.14 {"installer":{"name":"uv","version":"0.11.14","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
40a0164c5d6c3bfd8f10addbe6f9c7b2911b4ed115ee60ffb541bb0279cf336e
|
|
| MD5 |
cbb326527054ae69bd41408bf5c25c3e
|
|
| BLAKE2b-256 |
f70f300dc6e7c30edd404fce857a15829f6642213bcabb2b03fb34469a79e462
|