scikit-learn
torch
trimesh
roma
torchvision