scipy
torch
tonic
einops