wandb>=0.15.0
gitpython>=3.1.31
numpy>=1.23.5
lightning>=2.0.0
torch>=2.0.0
einops>=0.6.0
tqdm>=4.65.0
matplotlib>=3.7.1
pyyaml>=6.0
pydantic>=1.10.9
