numpy==1.20.3 torch==1.8.1+cu111 tokenizers==0.10.3 pytorch_lightning==1.3.1 regex==2021.4.4 transformers==4.7.0 tqdm==4.49.0 activations==0.1.0 dataclasses torchmetrics==0.5 file_utils==0.0.1 flax==0.3.4 PyYAML==5.4.1 utils==1.0.1 Cython faiss-gpu==1.7.2 tensorboard==2.8.0