torch
numpy
tqdm
matplotlib
scipy
drjit
rayd
smplpytorch
