keras>=3.0.0

[dev]
pytest>=7.0.0
build
twine

[docs]
mkdocs>=1.5.0
mkdocs-material>=9.0.0
mkdocstrings[python]>=0.20.0
mkdocs-autorefs
mkdocs-jupyter>=0.24.0

[jax]
jax[cpu]

[tensorflow]
tensorflow-cpu

[test]
pytest>=7.0.0

[torch]
torch
torchvision
