keon

jepa

keon

implementing minimal versions of joint-embedding predictive architecture (JEPA)

AI 简介

这是一个轻量级的JEPA(联合嵌入预测架构)算法教学实现项目,提供I-JEPA、V-JEPA、C-JEPA及LeWorldModel等主流变体的单文件PyTorch复现。核心特点包括:纯PyTorch实现、零外部依赖、配套图文教程、可视化支持(掩码网格、t-SNE/PCA演化、线性探针评估),并明确标注了与原论文在关键设计上的忠实性与教育性简化。适用于自监督表征学习、世界模型入门教学、视频/图像预测建模等研究与教学场景。

Python
221
Stars
23
Forks
98
Watchers
1
Issues

Star 增长

今日0
近 7 天0
近 30 天+9
综合评分42.04
默认分支main