jax>=0.4.26
numpy>=1.24

[dev]
build
nbclient
nbformat
ipykernel
pytest
twine

[docs]
matplotlib
myst-nb
roman
sphinx
sphinx-autodoc-typehints
sphinx-book-theme
sphinx-copybutton
sphinx-design
sphinx-togglebutton
sphinxcontrib-mermaid
