I have been using TensorFlow for coding up a number of different self-supervised models for vision and the experience have been great (and easy). Feel free to take a look into the following minimal implementations of popular self-supervised methods for vision:

SimCLR 6
SwAV 5
SimSiam 2

Sorry, could not post links because the forum does not yet allow it.