# Tensorflow.
tensorflow

# Torch.
# TODO: Use Torch CPU, remove after resolving Cuda version differences with TF
torch>=2.0.1+cpu
torchvision>=0.15.1

# Jax.
jax[cpu]

# Common deps.
-r requirements-common.txt