numpy
pandas
scikit-learn
scipy
torch
torchmetrics
torchvision
rich
typer
wandb
datasets
psutil
