| name | installation |
| description | How to install prime-rl and its optional dependencies. Use when setting up the project, installing extras like deep-gemm for FP8 models, or troubleshooting dependency issues. |
Installation
Basic install
uv sync
This installs all core dependencies defined in pyproject.toml.
All extras at once
The recommended way to install for most users:
uv sync --all-extras
This installs all optional extras (flash-attn, flash-attn-cute, etc.) in one go.
FP8 inference with deep-gemm
For certain models like GLM-5-FP8, you need deep-gemm. Install it via the fp8-inference dependency group:
uv sync --group fp8-inference
This installs the pre-built deep-gemm wheel. No CUDA build step is needed.
Dev dependencies
uv sync --group dev
Installs pytest, ruff, pre-commit, and other development tools.
Key files
pyproject.toml — all dependencies, extras, and dependency groups
uv.lock — pinned lockfile (update with uv sync --all-extras)