numpy>=1.26
tf_keras==2.19.0
ml_dtypes>=0.4.1
mplhep
matplotlib>=3.7
pillow>=10.0
scipy
pandas
hist

[:extra != "gpu"]
tensorflow<2.20,>=2.17
tensorflow-probability>=0.24

[cpu]
tensorflow<2.20,>=2.17
tensorflow-probability>=0.24

[dev]
pytest
pytest-cov
flake8

[docs]
sphinx>=5.0
sphinx-rtd-theme
sphinx-autodoc-typehints
sphinx-autoapi
myst-parser
pydata-sphinx-theme

[gpu]
tensorflow[and-cuda]<2.20,>=2.17
tensorflow-probability>=0.24
