-e ./
arviz
distrax
jax
jaxlib
jaxopt
myst_nb
optax
scikit-learn
seaborn
sphinx
sphinx-autobuild
sphinx-book-theme
sphinx-math-dollar
tensorflow
tensorflow-datasets
watermark

-e ./
arviz
distrax
jax
jaxlib
jaxopt
jupytext
myst_nb
optax
scikit-learn
seaborn
sphinx
sphinx-autobuild
sphinx-book-theme
sphinx-math-dollar
tensorflow
tensorflow-datasets
watermark

Add jupytext dependency to doc requirements
Add jupytext dependency to doc requirements
