einops
torch
numpy
wandb
pydantic
tqdm
