accelerate
datasets
jaxtyping>=0.3.3
ml_dtypes
natsort
numpy
peft>=0.17.0
pyarrow
simple-parsing
torch
transformers<4.56.0

[benchmarks]
wandb
kronfluence
dattri

[dev]
pre-commit
pytest
pyright
trl
furo
myst-parser
sphinx
sphinx-autodoc-typehints
sphinx-rtd-theme
nbsphinx
torch-tb-profiler>=0.4.3

[example]
backoff>=2.2.1
bitsandbytes>=0.49.0
pydantic>=2.12.5
trl

[faiss]
faiss-gpu-cu12
