tensorflow>=2.17.0
tensorflow-probability>=0.24.0
tf-keras>=2.17.0
numpy>=1.26.4
scipy>=1.14.0
matplotlib>=3.9.2

[cuda]
tensorflow[and-cuda]

[dev]
build
pytest-xdist
pytest
pytest-cov
ruff
twine

[metal]
tensorflow==2.13.1
tensorflow-metal==1.0.0
tensorflow-probability==0.21.0

[test]
pytest-xdist
pytest
pytest-cov
ruff
