beartype
einops
torch
tqdm
