JAX/Flax implementation of "VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learning". ICLR 2022. https://arxiv.org/abs/2105.04906
Primary LanguageJupyter NotebookMIT LicenseMIT