pytorch-lightning
kornia
gdown