torch
pytorch-lightning
simple-parsing
albumentations
matplotlib