torch>=2.0.0
transformers>=4.35.0
accelerate>=0.24.0
safetensors>=0.4.0
numpy>=1.24.0

[all]
cambium-llm[dev,train]
unsloth

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black==23.12.1
isort>=5.12.0
mypy>=1.5.0
pre-commit>=3.0.0
sphinx>=7.0.0
sphinx-rtd-theme>=2.0.0
myst-parser>=2.0.0

[train]
datasets>=2.14.0
trl>=0.7.0
wandb>=0.15.0
bitsandbytes>=0.41.0
