requirements.txt

torch==2.2.0
tensorflow==2.15.0
datasets==2.18.0
torchtext==0.17.0
torchvision==0.17.0
torchsummary==1.5.1
accelerate==0.26.0
gensim==4.3.2
transformers==4.39.3
pynvml==11.5.0
seqeval==1.2.2
triton==2.2.0
jupyter==1.0.0
jupyterlab-git==0.50.0
urllib3<2
scikit-learn
scipy==1.10.1
numpy
fastai==2.7.14
timm==0.9.12
tensorboard
albumentations==1.4.3
seaborn
tqdm==4.66.2
nbdime
matplotlib
opencv-python
Keras-Preprocessing==1.1.2
flask==2.1.0
Werkzeug==2.2.2
wandb==0.17.0