I created a minimal one-file implementations (160loc) of JEPA family (ijepa, vjepa, vjepa2, cjepa) for educational purposes [P]

Hi all,

I made my own minimal implementation of JEPA algorithms.

Making things minimal and removing all the things needed for scaling the algorithm always helped me understanding. So I stripped everything but the algorithm parts. What's left is 160-200 lines of code that distills the essence of the mathematics.

It is very easy to compare with the math in the paper and the code and how it can be implemented in PyTorch.

I added [algo]_tutorial.md files to help with understanding.

https://github.com/keon/jepa