torch
torcheval
pybigtools
numpy
pandas
tqdm
huggingface_hub

[dev]
pytest
wandb
matplotlib
scikit-learn
