torch
gradio
einops
scipy
sklearn