scikit-learn
transformers
codecarbon
datasets
torch
pandas
wandb
pyyaml
tlr