torch
torchvision
selavi @ git+https://github.com/YingtianDt/selavi.git