torch
numpy
tensorboardx
distributed_worker>=1.3.2
