torch>=2.1.0
transformers>=4.40.0
accelerate>=0.27.0

[all]
bitsandbytes>=0.43.0
peft>=0.10.0
trl>=0.8.0
datasets>=2.18.0
triton>=2.3.0
deepspeed>=0.14.0
optuna>=3.6.0

[deepspeed]
deepspeed>=0.14.0

[dev]
pytest>=8.0
pytest-cov>=5.0
black>=24.0
ruff>=0.4.0
mypy>=1.10
twine>=5.0
build>=1.2

[flash]
flash-attn>=2.5.0

[full]
bitsandbytes>=0.43.0
peft>=0.10.0
trl>=0.8.0
datasets>=2.18.0

[optuna]
optuna>=3.6.0

[triton]
triton>=2.3.0
