torch>=1.9.0
numpy>=1.19.5
wandb>=0.12.5
scikit-learn>=0.24.1
transformers>=4.5.1
einops>=0.3.0
datasets>=1.6.2
tqdm>=4.49.0
blitz-bayesian-pytorch>=0.2.7
gpytorch>=1.5.0
scipy>=1.5.4
dill>=0.3.3
joblib>=1.0.1
alpaca-ml>=0.8.2
