einops
numpy
pandas
pytorch_lightning>=1.5
PyYAML
scikit_learn
scipy
tables
test_tube
torch>=1.9
torch_geometric>=2.0
torchmetrics>=0.7
tqdm

[full]
matplotlib
mpld3
neptune-client>=0.14
pytorch_fast_transformers
