wehub-resource-sync
27 lines
619 B
Plaintext
27 lines
619 B
Plaintext
# Tensorflow.
|
|
# Note: when the version of Tensorflow is changed, the version tf_keras must be
|
|
# changed in .github/workflows/actions.yml (pip install --no-deps tf_keras).
|
|
tensorflow-cpu;sys_platform != 'darwin'
|
|
tensorflow;sys_platform == 'darwin'
|
|
ai-edge-litert
|
|
|
|
# Torch.
|
|
--extra-index-url https://download.pytorch.org/whl/cpu
|
|
torch
|
|
torch-xla;sys_platform != 'darwin'
|
|
|
|
# Jax.
|
|
# Pinned to 0.8.0 on CPU for CI compatibility with older backends.
|
|
# Note that we test against the latest JAX on GPU.
|
|
jax[cpu]
|
|
flax
|
|
|
|
# OpenVino
|
|
openvino
|
|
|
|
# Common deps.
|
|
-r requirements-common.txt
|
|
|
|
# LiteRT torch converter.
|
|
litert-torch>=0.9.1
|