File size: 204 Bytes
17c6d62 |
1 2 3 4 5 6 7 8 9 |
jax>=0.2.8 jaxlib>=0.1.59 flax>=0.3.5 optax>=0.0.8 -f https://download.pytorch.org/whl/torch_stable.html torch==2.2.0 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.12.0+cpu |