jax
jaxlib
tqdm

[docs]
furo==2020.12.30b24
nbsphinx==0.8.1
nb-black==1.0.7
matplotlib==3.3.3
sphinx-copybutton==0.3.5
Sphinx==4.4.0
isort==5.10.1
sphinx-rtd-theme==1.0.0

[tests]
pytest
