joint-embedding-predictive-architecture

Here are 10 public repositories matching this topic...

galilai-group / stable-pretraining

Reliable, minimal and scalable library for pretraining foundation and world models

computer-vision deep-learning transformers pytorch distributed computer-vision-algorithms multimodal-learning self-supervised-learning contrastive-learning joint-embedding foundation-models large-language-model joint-embedding-predictive-architecture stable-pretraining

Updated Sep 29, 2026
Python

geriskenderi / graph-jepa

Star

Official code for the paper Graph-level Representation Learning with Joint-Embedding Predictive Architectures published in Transactions on Machine Learning Research (TMLR)

self-supervised-learning graph-representation-learning graph-neural-network joint-embedding-predictive-architecture

Updated Jan 17, 2025
Python

ayoubncbae / awesome-medical-world-models

Star

Curated papers, code, datasets, and benchmarks for medical world models in imaging, EHR trajectories, treatment planning, surgical AI, robotics, and virtual-cell simulation.

awesome-list dreamer world-models medical-ai clinical-ai healthcare-ai counterfactual-simulation foundation-models predictive-simulation jepa joint-embedding-predictive-architecture latent-dynamics rssm latent-world-models medical-world-models clinical-world-models generative-world-models generative-latent-prediction predictive-representation-learning

Updated Jun 4, 2026
TeX

Griffith-7 / H-JEPA-LM

Star

A PyTorch implementation of a Hierarchical Joint-Embedding Predictive Language Model (H-JEPA-LM) featuring action conditioning and latent-space world model planning.

machine-learning pytorch language-model self-supervised-learning world-models joint-embedding jepa joint-embedding-predictive-architecture

Updated Aug 3, 2026
Python

Mattral / Worldgap

Star

Reusable world-model-based domain-gap quantification — for perception pipelines and actuator/mechanism models — before any hardware is touched.

representation-learning robotics-perception world-models joint-embedding-predictive-architecture

Updated Oct 15, 2026
Python

nhatkhangcs / tijepa

Star

Official codebase for TI-JEPA, the Text-Image Joint-Embedding Predictive Architecture. First outlined in our Capstone Project Defense, got 9.9/10

python nlp transformer multimodal soict multimodal-fusion joint-embedding-predictive-architecture

Updated Jul 21, 2025
Python

a controlled comparison study of SSL masked-modeling objectives where masked spectrograms reconstruction MAE, masked latent prediction M2D, and joint-embedding prediction Audio-JEPA are pretrained on 52h corpus of Indic speech across 12 languages, all sharing an identical ViT-S/16 encoder and evaluated on the LID & SID tasks of IndicSUPERB

masked-autoencoder audio-self-supervised-learning masked-modeling-duo joint-embedding-predictive-architecture