
[pytorch]
torch>=1.10.1
torchvision>=0.11.2
torchaudio>=0.10.1
pytorch-lightning>=1.6.5
simplejpeg
albumentations

[tensorflow]
tensorflow>=2.10.0
protobuf<=3.19
