torch==2.1.1
torchdata==0.7.1
alive_progress==3.1.5