torch>=2.0.0
numpy>=1.20.0
tqdm>=4.60.0
pyyaml>=6.0
transformers>=4.57.1
datasets>=2.0.0

[dev]
pytest>=7.0.0
black>=22.0.0
flake8>=4.0.0
mypy>=0.950
mkdocs>=1.5.0
mkdocs-material>=9.0.0
mkdocstrings[python]>=0.24.0
build
twine

[flash]
flash-attn>=2.0.0

[wandb]
wandb>=0.16.0
