torch==2.5.1
numpy==1.22.0
einops==0.8.0
transformers==4.47.1
matplotlib==3.8.4
tqdm==4.67.1
pytest==8.3.4

[dev]
pytest>=8.3.4
black>=22.0.0
isort>=5.10.0
flake8>=4.0.0
