torch
torchvision
torchmetrics
lightning
matplotlib
gradio
termcolor