solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning

Framework for training and evaluating self-supervised learning methods for speaker verification.

This a tensorflow implementation of VICReg - a self-supervised learning architecture that prevents collapse in an intuitive manner using a loss function that 1. maintains the variance of each embedding over a batch above a threshold and 2. decorrelating pairs of embeddings over a batch and attracting them to 0. Training was done using TPU on colab

A VIcReg Implementation in pytorch

Train a JEPA world model on a set of pre-collected trajectories from a toy environment involving an agent in two rooms.