torch
torchvision
numpy
transformers
matplotlib
gym
