torch
gradio
pydantic
PyYAML
transformers
scikit-learn
flash_attn
adam-atan2
einops
tqdm
coolname
pydantic
argdantic
wandb
omegaconf
hydra-core
huggingface_hub