https://blog.tnichols.org/posts/contrastive-learning/