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

[all]
peft>=0.10.0
bitsandbytes>=0.43.0
trl>=0.8.0
datasets>=2.18.0
wandb
tensorboard
deepspeed
triton

[dev]
pytest>=8.0
black
ruff
twine
build

[distributed]
deepspeed

[flash]
flash-attn>=2.5.0

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

[logging]
wandb
tensorboard

[triton]
triton
