Installation¶
Requirements¶
Python 3.11 or later
JAX (with optional GPU support)
Install with pip¶
pip install prosemble
Install with uv¶
uv add prosemble
GPU Support¶
For GPU acceleration, install JAX with CUDA support first:
# For CUDA 12
pip install jax[cuda12]
# Then install prosemble
pip install prosemble
Development Installation¶
git clone https://github.com/naotoo1/prosemble.git
cd prosemble
uv sync --all-extras
# Or with devenv (Nix)
devenv shell
Verify Installation¶
>>> import prosemble
>>> print(prosemble.__version__)
2.0.0
>>> import jax
>>> print(jax.devices()) # Check available devices