torch>=2.2.0
torchvision>=0.17.0
accelerate>=1.0.0
transformers[kernels]>=4.46.0
huggingface-hub>=0.24.0
bitsandbytes>=0.44.0
peft>=0.13.0
optuna>=4.0.0
numpy>=1.24.0
datasets>=2.18.0
tqdm>=4.66.0
psutil>=5.9.0
py-cpuinfo>=9.0.0
rich>=13.7.0
questionary>=2.0.0
pydantic>=2.0.0
pydantic-settings>=2.0.0
tomli-w>=1.0.0
nvidia-ml-py>=12.0.0
requests<3.0,>=2.32.0
chardet<6.0,>=5.0.0

[dev]
ruff>=0.14.5
pytest>=7.0.0

[lmeval]
lm-eval[hf]<0.5,>=0.4.9

[native]
datasets>=2.18.0

[research]
geom-median>=1.2
scikit-learn>=1.3
pacmap>=0.8
imageio>=2.31
matplotlib>=3.8
