jepa

该项目实现了联合嵌入预测架构（JEPA）的最小版本。它使用Python和PyTorch框架，提供了I-JEPA、V-JEPA及其变体等算法的独立实现，并附带详细的教程。每个算法文件仅依赖于`torch`和`torchvision`库，通过额外的可视化脚本可生成如掩码网格、损失曲线及线性探针等分析图表。适用于需要进行自监督学习或表示学习的研究者与开发者，在CIFAR-10、Moving MNIST等数据集上测试模型性能或探索世界模型时尤为有用。