torch==2.0.1
torchvision==0.15.2
numpy
tqdm
declearn
scikit-learn
pandas
matplotlib

[energy]
