#jax[tpu]>=0.2.16 -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
flax
joblib
zstandard
torch
tensorflow
tensorboard
transformers
datasets
wandb