add requirements
Browse files- requirements.txt +4 -0
requirements.txt
CHANGED
|
@@ -1,6 +1,10 @@
|
|
|
|
|
| 1 |
flax
|
|
|
|
|
|
|
| 2 |
torch
|
| 3 |
tensorflow
|
|
|
|
| 4 |
transformers
|
| 5 |
datasets
|
| 6 |
wandb
|
|
|
|
| 1 |
+
jax[tpu]>=0.2.16 -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
|
| 2 |
flax
|
| 3 |
+
joblib
|
| 4 |
+
zstandard
|
| 5 |
torch
|
| 6 |
tensorflow
|
| 7 |
+
tensorboard
|
| 8 |
transformers
|
| 9 |
datasets
|
| 10 |
wandb
|