| #-f https://download.pytorch.org/whl/torch_stable.html | |
| -f https://download.pytorch.org/whl/cu116 | |
| -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html | |
| protobuf<3.20 | |
| pandas | |
| torch | |
| transformers>=4.13.0 | |
| langdetect | |
| psutil | |
| jax[cuda]==0.3.16 | |
| chex>=0.1.4 | |
| ##jaxlib==0.1.67 | |
| flax>=0.5.3 | |
| sentencepiece | |