jax>=0.8.0
jaxlib>=0.4.0
jaxtyping>=0.2.23
beartype>=0.14.0

[dev]
black>=24.8.0
isort>=5.13.2
pydoclint>=0.6.0
pytest>=8.3.0
pytest-cov>=5.0.0
