pygame==2.1.2
torch
torchvision

scikit-learn==1.0.2
pandas==1.4.2

pytorch_lightning
matplotlib