Installation

Install the most recent stable version

Jaxley is available on PyPI:

pip install jaxley

This will install Jaxley with CPU support. If you want GPU support, follow the instructions on the JAX github repository to install JAX with GPU support (in addition to installing Jaxley). For example, for NVIDIA GPUs, run

pip install -U "jax[cuda12]"

Install from source

You can also install Jaxley from source:

git clone https://github.com/jaxleyverse/jaxley.git
cd jaxley
pip install -e .

Note that pip>=21.3 is required to install the editable version with pyproject.toml see pip docs.