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

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

[example]
trl

[faiss]
faiss-gpu-cu12
