keras-team--keras
wehub-resource-sync
14 行
322 B
Plaintext
14 行
322 B
Plaintext
# Tensorflow cpu-only version (needed for testing).
|
|
tensorflow-cpu
|
|
|
|
# Torch cpu-only version (needed for testing).
|
|
--extra-index-url https://download.pytorch.org/whl/cpu
|
|
torch
|
|
|
|
# Jax with cuda support.
|
|
--find-links https://storage.googleapis.com/jax-releases/libtpu_releases.html
|
|
jax[tpu]
|
|
flax
|
|
|
|
-r requirements-common.txt
|