keras>=3.0.0
numpy

[dev]
pytest
tensorflow
tf_keras
jax
torch

[jax]
jax
jaxlib

[tensorflow]
tensorflow

[torch]
torch
