14 lines
243 B
Plaintext
14 lines
243 B
Plaintext
# Tensorflow.
|
|
tf-nightly-cpu==2.15.0.dev20231009 # Pin a working nightly until rc0.
|
|
|
|
# Torch.
|
|
--extra-index-url https://download.pytorch.org/whl/cpu
|
|
torch>=2.1.0
|
|
torchvision>=0.16.0
|
|
|
|
# Jax.
|
|
jax[cpu]
|
|
|
|
# Common deps.
|
|
-r requirements-common.txt
|