numpy
tqdm
torch
matplotlib
muon
pyro-ppl
