numpy
scikit-learn
scipy
keras>=3.0

[all]
mcfly[dev,jax,publishing,tensorflow,torch]

[dev]
coverage
prospector[with_pyroma]
pytest
pytest-cov

[jax]
jax>=0.4.20

[publishing]
build
twine
wheel
sphinx
sphinx-rtd-theme

[tensorflow]
tensorflow>=2.16.1

[torch]
torch>=2.1.0
