ipython
jupyter
matplotlib
torchmetrics