torch>=1.10.0
rich
tensorboard
