accelerate
datasets
jinja2
jaxtyping>=0.3.3
ml_dtypes
natsort
numpy
scipy
peft>=0.17.0
petname
pyarrow
pyyaml
simple-parsing
torch<2.10,>=2.5
torchvision<0.25,>=0.20
torchaudio<2.10,>=2.5
torchopt
transformers>=5.0
bitsandbytes
huggingface-hub>=1.13.0

[benchmarks]
wandb
kronfluence
dattri

[dev]
pre-commit
pytest
pyright==1.1.378
trl
furo
myst-parser
scipy-stubs
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
wandb
lm_eval

[faiss]
faiss-gpu-cu12
