numpy>=1.21

[docs]
sphinx
pydata-sphinx-theme
myst-parser

[jax]
jax>=0.4
