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"
  ☆123Updated 7 months ago
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆84Updated 10 months ago
• alexiglad / EBT
  PyTorch Code for Energy-Based Transformers paper -- generalizable reasoning and scalable learning
  ☆553Updated this week
• m1balcerak / EnergyMatching
  [NeurIPS 2025] Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling"
  ☆174Updated 3 weeks ago
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆111Updated 3 weeks ago
• facebookresearch / jepa-intuitive-physics
  This repo contains the code for the paper "Intuitive physics understanding emerges fromself-supervised pretraining on natural videos"
  ☆196Updated 8 months ago
• Chrixtar / SRM
  [ICML 2025] Implementation of Spatial Reasoning with Denoising Models
  ☆83Updated 3 months ago
• zacharyhorvitz / Fk-Diffusion-Steering
  A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.
  ☆194Updated 4 months ago
• raywang4 / EqM
  ☆157Updated last month
• ChenWu98 / algorithmic-creativity
  [ICML 2025] Roll the dice & look before you leap: Going beyond the creative limits of next-token prediction
  ☆74Updated 5 months ago
• lucidrains / hl-gauss-pytorch
  The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch
  ☆67Updated 3 months ago
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆105Updated 3 weeks ago
• facebookresearch / Mixture-of-Transformers
  Mixture-of-Transformers: A Sparse and Scalable Architecture for Multi-Modal Foundation Models. TMLR 2025.
  ☆121Updated 2 months ago
• bluorion-com / ZClip
  Official implementation of the paper: "ZClip: Adaptive Spike Mitigation for LLM Pre-Training".
  ☆138Updated last month
• rbalestr-lab / llm-jepa
  ☆122Updated last month
• apple / ml-tarflow
  ☆296Updated 10 months ago
• lucidrains / spline-based-transformer
  Implementation of the proposed Spline-Based Transformer from Disney Research
  ☆105Updated last year
• fal-ai / diffusion-speedrun
  Focused on fast experimentation and simplicity
  ☆75Updated 10 months ago
• apple / ml-flextok
  FlexTok: Resampling Images into 1D Token Sequences of Flexible Length
  ☆267Updated 5 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
• kvfrans / splus
  ☆119Updated 5 months ago
• lucidrains / dreamer4
  Implementation of Danijar's latest iteration for his Dreamer line of work
  ☆106Updated this week
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆96Updated 3 months ago
• google-deepmind / md4
  Official Jax Implementation of MD4 Masked Diffusion Models
  ☆139Updated 8 months ago
• apple / ml-l3m
  Large multi-modal models (L3M) pre-training.
  ☆221Updated last month
• lucidrains / nGPT-pytorch
  Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI
  ☆291Updated 5 months ago
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆106Updated last year
• lucidrains / evolutionary-policy-optimization
  Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University
  ☆102Updated last month
• lucidrains / genie2-pytorch
  Implementation of a framework for Genie2 in Pytorch
  ☆153Updated 10 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆294Updated last year