tensorflow>=2.15.0
tensorflow-probability>=0.23.0
numpy
scipy
matplotlib

[cuda]
tensorflow[and-cuda]

[dev]
pyarrow
pytest
pytest-cov
pylint
pylint[spelling]
twine

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