Development Guide

Crente venv:

python3.12 -m venv venv
. venv/bin/activate

Install the latest version of Mypy. This is because of the python/mypy#15238 issue.

pip install -U pip
pip install -U wheel
pip install -U git+https://github.com/python/mypy.git

Install dependencies:

pip install "jax[cpu]"
pip install -r requirements.txt
pip install -r docs/requirements.txt

Run test:

python tests/test_einshard.py

Build package:

pip install build
python -m build

Build docs:

cd docs
make html

cd docs/_build/html
python -m http.server -b 127.0.0.1