matplotlib
numpy
wandb