Installation¶

Prerequisites¶

Python 3.12
uv package manager

Install from PyPI¶

Samudra is pure Python, so one wheel covers every platform. The GPU custom kernels are opt-in:

# Install with `uv` (recommended)
uv add samudra                    # CPU (default)
uv add "samudra[cuda]"            # adds flash-attn, flash-perceiver, torchvision
uv add samudra --prerelease=allow # latest nightly dev build
# Install with `pip`
pip install samudra               # CPU (default)
pip install "samudra[cuda]"       # adds flash-attn, flash-perceiver, torchvision
pip install --pre samudra         # latest nightly dev build

The cuda extra compiles native kernels against your local CUDA + torch; see Releasing to PyPI for the details.

Development setup¶

To work on Samudra itself, clone the repository and install dependencies:

git clone https://github.com/m2lines/Samudra.git
cd Samudra
uv sync --dev
source .venv/bin/activate

Verify Installation¶

Print the training CLI help to confirm everything is set up correctly:

uv run -m samudra.train --help