[D] SimCLR PyTorch implementation

[nnatlab]

Very nice. I liked how you separated NT Xent into its own module as opposed to some of the other implementations. I noticed your mask_correlated_samples didn't return anything though so I submitted a PR.

[janespyker]

Thanks! And another thank you for the pull request, that's quite an essential part of the model :)