torch
tensorboard>=2.3.0
six
werkzeug
numpy
