wandb
pyhessian
sklearn
matplotlib
pytorch
transformers