requirements.txt

# PyTorch.
torch==1.10.1+cu113
torchvision==0.11.2+cu113
torchaudio==0.10.1+cu113

# PyTorch Geometric.
torch-scatter==2.0.9 
torch-sparse==0.6.13 
torch-cluster==1.6.0 
torch-spline-conv==1.2.1 
torch-geometric==2.0.4

# Hugging Face.
transformers==4.16.2
tokenizers==0.11.4
datasets==1.18.3

# Optimization.
deepspeed==0.5.10 #NB: Change line 1717 of 'deepspeed/runtime/zero/stage_1_and_2.py' from "bit16_partitions[partition_id]" to "[bit16_partitions[partition_id]]".
git+https://github.com/luyug/GradCache@main#egg=GradCache

# Visualisation.
matplotlib==3.2.2
seaborn==0.11.2  
plotly==5.8.0
kaleido==0.2.1
pyvis==0.2.0

# Other (ML).
numpy==1.21.0
scikit-learn==1.0.2
pytorch-lightning==1.5.10
sentence-transformers==2.1.0
wandb==0.12.10
networkx==2.6.3
node2vec==0.4.3
sentencepiece==0.1.97
easynmt==2.0.2

# Other (data).
pandas==1.3.5
spacy==3.3.0
fr_core_news_md @ https://github.com/explosion/spacy-models/releases/download/fr_core_news_md-3.3.0/fr_core_news_md-3.3.0-py3-none-any.whl

# Other.
jupyter==1.0.0

# Package locations.
--find-links https://download.pytorch.org/whl/cu113/torch_stable.html
--find-links https://data.pyg.org/whl/torch-1.10.0+cu113.html