# Tensorflow
tensorflow-cpu>=2.13.0

# Torch.
torch>=2.1.0+cpu
torchvision>=0.16.0+cpu

# Jax.
jax[cpu]

# Common deps.
-r requirements-common.txt