jax[cuda]>=0.8.0
jaxlib>=0.4.0
jaxtyping>=0.2.23
beartype>=0.14.0
yggdrax<0.1.0,>=0.0.1

[dev]
black>=24.8.0
isort>=5.13.2
pytest>=8.3.0
pytest-cov>=5.0.0
pytest-xdist>=3.6.0
pre-commit>=3.8.0
build>=1.2.2
twine>=5.1.1
