# For CI
tensorflow-cpu

--extra-index-url https://download.pytorch.org/whl/cpu
torch
torchvision

jax[cpu]

keras