jax==0.1.35
jaxlib>=0.1.14
tqdm

[dev]
ipython

[doc]
sphinx
sphinx_rtd_theme

[examples]
matplotlib

[test]
flake8
pytest>=4.1
