gemmi
matplotlib
numpy
numba
onnxruntime
pandas
scikit-learn
scipy

[dataset_generation]
pyarrow
openpyxl
tqdm

[mpi]
mpi4py

[training]
pyarrow
openpyxl
skl2onnx
keras
torch
torchvision
