albumentations
iglovikov_helper_functions
pytorch_lightning
pytorch_toolbelt
timm
torch
tqdm
wandb

[test]
pytest
