14 lines
211 B
Plaintext
14 lines
211 B
Plaintext
# 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
|