torch
numpy
einops
safetensors
simple_parsing
huggingface_hub

[dev]
pytest
build
twine

[otf]
pandas
pylate

[train]
wandb
datasets
pyarrow
pandas
tqdm
pyyaml
bitsandbytes

[widgets]
ipyreact
