numpy>=1.13.0
scipy>=1.0.0
matplotlib>=2.2.0
matplotlib-scalebar>=0.5.0
tqdm>=4.66.0
jax[cuda,cuda11,cuda12]>=0.4.17
jaxtyping>=0.2.21
chex>=0.1.85
optax>=0.1.8
pytest>=8.0.0
beartype>=0.19.0
black[jupyter]>=25.1.0

[docs]
ipykernel>=6.29.5
myst-parser>=4.0.1
nbsphinx>=0.9.6
pydata-sphinx-theme>=0.16.1
sphinx>=8.1.3
sphinx-rtd-theme>=3.0.2
sphinxcontrib-napoleon>=0.7
