click==8.1.7
torch==2.1.2
scipy==1.10.1
wandb==0.16.2
protobuf==3.20.0
urllib3<2,>=1.26.18
pydantic==2.6.0
pydantic-yaml==1.2.0
ray[default]==2.9.1
datasets==2.16.1
transformers==4.36.2
accelerate==0.26.1
peft==0.7.1
trl==0.7.10
bitsandbytes==0.42.0
lm-eval[openai]==0.4.1
einops==0.7.0

[dev]
lm-buddy[docs,ruff,test]
pre-commit==3.6.0
jupyter==1.0.0

[docs]
Sphinx==7.2.6
nbsphinx==0.9.3
sphinx-codeautolink==0.15.0
myst-parser==2.0.0
furo==2024.1.29

[ruff]
ruff==0.2.1

[test]
pytest==7.4.3
pytest-cov==4.1.0
