Alternatives and similar repositories for FlexMDM

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

• google-deepmind / md4
  Official Jax Implementation of MD4 Masked Diffusion Models
  ☆126Updated 6 months ago
• yilundu / ired_code_release
  ☆70Updated last year
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆86Updated last week
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆101Updated last month
• ChenWu98 / algorithmic-creativity
  [ICML 2025] Roll the dice & look before you leap: Going beyond the creative limits of next-token prediction
  ☆68Updated 3 months ago
• AllanYangZhou / nfn
  NF-Layers for constructing neural functionals.
  ☆88Updated last year
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆104Updated last year
• lucidrains / self-reasoning-tokens-pytorch
  Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto
  ☆56Updated last year
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆58Updated 4 months ago
• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆48Updated 2 years ago
• bm371613 / conditional-flow-matching
  ☆14Updated last year
• kylehkhsu / latent_quantization
  ☆41Updated last year
• microsoft / ReinMax
  Beyond Straight-Through
  ☆102Updated 2 years ago
• wmn-231314 / diffusion-data-constraint
  Official PyTorch implementation and models for paper "Diffusion Beats Autoregressive in Data-Constrained Settings". We find diffusion mod…
  ☆95Updated 3 weeks ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆35Updated last year
• yukara-ikemiya / modified-shortcut-models-pytorch
  PyTorch implementation of Shortcut Models [Frans, 2025] with little modification
  ☆53Updated 2 months ago
• apple / ml-agm
  ☆44Updated last year
• ozekri / SEPO
  Code for the paper: "Fine-Tuning Discrete Diffusion Models with Policy Gradient Methods"
  ☆27Updated 4 months ago
• lebellig / flow-matching
  Annotated Flow Matching paper
  ☆208Updated last year
• lucidrains / coordinate-descent-attention
  Implementation of an Attention layer where each head can attend to more than just one token, using coordinate descent to pick topk
  ☆46Updated 2 years ago
• NanboLi / FACTS
  [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"
  ☆28Updated 3 months ago
• yilundu / reduce_reuse_recycle
  ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC
  ☆145Updated 11 months ago
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆73Updated 8 months ago
• zacharyhorvitz / Fk-Diffusion-Steering
  A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.
  ☆186Updated 2 months ago
• google-deepmind / spectral_ssm
  ☆34Updated last year
• watcl-lab / positional_attention
  Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"
  ☆14Updated 3 months ago
• radarFudan / mamba-minimal-jax
  ☆34Updated 10 months ago
• klieret / wandb-offline-sync-hook
  A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!
  ☆85Updated 3 weeks ago
• kvfrans / splus
  ☆120Updated 3 months ago
• singhgautam / slate
  This is the official source code for SLATE. We provide the code for the model, the training code, and a dataset loader for the 3D Shapes …
  ☆86Updated 2 years ago