torch
numpy
wandb
optuna
pyyaml
datasets
pyarrow
tqdm
dataclasses
simple_parsing
einops
safetensors
accelerate
transformers
bitsandbytes
huggingface_hub
ipyreact
