autodocsumm
myst-parser
sphinx
sphinx-rtd-theme
sympy

ipykernel
matplotlib
plotly
jupyter-sphinx
matscipy
jraph
nequip_jax @ git+https://github.com/mariogeiger/nequip-jax@1.1.0
flax
dm-haiku
equinox