albumentations
iglovikov_helper_functions
pytorch_lightning
pytorch_toolbelt
timm

[test]
pytest
