pygame
torch
numpy
