implementing minimal versions of joint-embedding predictive architecture (JEPA) - View it on GitHub