Alternatives and similar repositories for lejepa

Users that are interested in lejepa are comparing it to the libraries listed below

• facebookresearch / capi
  Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"
  ☆124Updated 7 months ago
• facebookresearch / jepa-intuitive-physics
  This repo contains the code for the paper "Intuitive physics understanding emerges fromself-supervised pretraining on natural videos"
  ☆203Updated 9 months ago
• alexiglad / EBT
  PyTorch Code for Energy-Based Transformers paper -- generalizable reasoning and scalable learning
  ☆562Updated 3 weeks ago
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆85Updated 10 months ago
• goombalab / hnet
  H-Net: Hierarchical Network with Dynamic Chunking
  ☆788Updated 2 weeks ago
• google-deepmind / tips
  ☆107Updated 7 months ago
• ShadeAlsha / ICon
  ICLR 2025 - official implementation for "I-Con: A Unifying Framework for Representation Learning"
  ☆117Updated 5 months ago
• apple / ml-flextok
  FlexTok: Resampling Images into 1D Token Sequences of Flexible Length
  ☆274Updated 6 months ago
• facebookresearch / Mixture-of-Transformers
  Mixture-of-Transformers: A Sparse and Scalable Architecture for Multi-Modal Foundation Models. TMLR 2025.
  ☆129Updated 2 months ago
• microsoft / dion
  Dion optimizer algorithm
  ☆395Updated 2 weeks ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆297Updated last year
• lucidrains / nGPT-pytorch
  Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI
  ☆294Updated 6 months ago
• bluorion-com / ZClip
  Official implementation of the paper: "ZClip: Adaptive Spike Mitigation for LLM Pre-Training".
  ☆139Updated 2 weeks ago
• helblazer811 / Diffusion-Explorer
  Interactive visualizations of the geometric intuition behind diffusion models.
  ☆903Updated 5 months ago
• apple / ml-l3m
  Large multi-modal models (L3M) pre-training.
  ☆222Updated 2 months ago
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆116Updated last month
• lucidrains / evolutionary-policy-optimization
  Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University
  ☆102Updated 2 months ago
• iShohei220 / adopt
  Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"
  ☆428Updated 11 months ago
• lorenmt / clarity-template
  Clarity: A Minimalist Website Template for AI Research
  ☆167Updated 10 months ago
• m1balcerak / EnergyMatching
  [NeurIPS 2025] Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling"
  ☆178Updated last month
• helblazer811 / ConceptAttention
  ConceptAttention: A method for interpreting multi-modal diffusion transformers.
  ☆354Updated 3 weeks ago
• apple / ml-tarflow
  ☆303Updated 11 months ago
• cloneofsimo / scaling-guide
  WIP
  ☆93Updated last year
• sayakpaul / nanoDiT
  Just another reasonably minimal repo for class-conditional training of pixel-space diffusion transformers.
  ☆138Updated 6 months ago
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆97Updated 4 months ago
• huggingface / ai-deadlines
  ⏰ AI conference deadline countdowns
  ☆290Updated last week
• rbalestr-lab / llm-jepa
  ☆137Updated 2 months ago
• lucidrains / titok-pytorch
  Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"
  ☆181Updated last year
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆186Updated last year
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆192Updated 8 months ago