[P] Choosing a self-supervised learning framework that's easy to use

I need to pretrain a spatial image classification model using self-supervised learning and would like to find the best/easiest to use implementation of some SOTA. There are a couple of factors:

• Tensorflow is preferred, but if there are great implementations in Pytorch, that is also fine
• Training on multiple GPUs needs to be supported

Also, these likely don't impact the framework selection, but worth noting:

• ResNet will be used as the (initial) backbone
• The dataset is huge, 1B images can …

framework learning machinelearning self-supervised learning supervised learning