black==24.3.0
click==8.1.7
einops==0.7.0
matplotlib==3.8.4
numpy==1.26.4
pytest==8.1.1
python-dotenv==1.0.1
requests==2.31.0
torch==2.2.2
tqdm==4.66.2
wandb==0.16.6