numpy
scikit-learn
rich
tensorboard
torch
