jax>=0.4.25
beartype>=0.20.0
jaxtyping>=0.2.38
einops>=0.8.1

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=23.0.0
isort>=5.12.0
mypy>=1.0.0
ruff>=0.9.0
pre-commit>=4.0.0

[docs]
sphinx>=6.0.0
sphinx-rtd-theme>=1.2.0
myst-parser>=2.0.0

[examples]
matplotlib>=3.8.0
