einops==0.6.1
matplotlib>=3.7.4
numpy==1.23.5
torch==2.0.1
tqdm==4.65.0

[dev]
pytest==7.4.3
