torch
hydra-core
omegaconf
graphein
torch_geometric
click>=8.1
wandb
accelerate
tokenizers
numpy
pandas
biopython
transformers
x-transformers

[:python_version < "3.11"]
typing-extensions
