matplotlib
pandas
pydantic
pytest
scipy
tqdm
wandb==0.21.1

einops
transformers

jupyter
jupyterlab
notebook
