torch>=2.7.0
transformers<6,>=4.48.3
datasets<6,>=2.14.0
accelerate<2,>=0.24.0
peft<1,>=0.6.0
bitsandbytes>=0.41.0
trl<2,>=0.12
numpy>=1.21.0
pyyaml>=6.0
huggingface-hub>=0.17.0
tqdm>=4.64.0
psutil>=5.9.0
requests>=2.28.0
pandas>=1.3.0
scikit-learn>=1.0.0
click>=8.0.0
rich>=12.0.0
typer>=0.7.0
wandb>=0.15.0
tensorboard>=2.12.0
evaluate>=0.4.0
rouge-score>=0.1.2
sacrebleu>=2.3.0
lm-eval>=0.4.8
langdetect>=1.0.9
immutabledict>=2.0.0
nltk>=3.8.0

[:platform_system != "Darwin"]
nvidia-ml-py>=11.495.46

[dev]
black>=23.0.0
isort>=5.12.0
flake8>=6.0.0
mypy>=1.5.0
pytest>=7.0.0
pytest-cov>=4.0.0
pytest-xdist>=3.0.0
pytest-mock>=3.10.0
pytest-timeout>=2.1.0
hypothesis>=6.70.0
pre-commit>=3.0.0
gpustat>=1.1.1

[dev:platform_system != "Darwin"]
nvitop>=1.1.0

[docs]
mkdocs>=1.5.0
mkdocs-material>=9.5.0
mike>=1.1.0
pymdown-extensions>=10.0.0
pygments>=2.15.0
mkdocstrings[python]>=0.24.0

[interpret]
transformer_lens>=2.0.0

[text-metrics]
bert-score>=0.3.13
sentence-transformers>=2.2.0
codebleu>=0.1.0

[unsloth]
unsloth>=2024.0

[viz]
matplotlib>=3.5.0
seaborn>=0.11.0

[vllm]
vllm>=0.5.0

[vllm-lens]
vllm>=0.5.0
vllm-lens>=0.1.0
