absl-py
sphinx >=3
sphinx-autodoc-typehints
sphinx-book-theme>=0.3.3
sphinx-copybutton>=0.5.0
sphinx-remove-toctrees
jupyter-sphinx>=0.3.2
myst-nb

# Packages used for CI tests.
pytest
pytest-xdist

# Packages used for notebook execution
matplotlib
scikit-learn
numpy
.[ci]  # Install jax from the current directory; jaxlib from pypi.