active_learn
scikit-learn
torchvision
wandb
