requests>=2.28.0
tqdm>=4.62.0
pyyaml>=6.0
rich>=13.0.0
click>=8.0.0
numpy>=1.21.0

[all]
langtrain[kernels,train,vision]
langtune>=0.1.41
langvision>=0.1.57

[kernels]
triton>=2.1.0
flash-attn>=2.3.0

[train]
torch>=2.0.0
transformers>=4.40.0
peft>=0.10.0
datasets>=2.14.0
accelerate>=0.28.0
bitsandbytes>=0.43.0
trl>=0.8.0
scipy>=1.10.0
scikit-learn>=1.3.0
pandas>=2.0.0
pyarrow>=12.0.0
openpyxl>=3.1.0

[vision]
torchvision>=0.15.0
Pillow>=10.0.0
langvision>=0.1.57
