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
  ☆151Updated 10 months ago
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆106Updated 2 months ago
• liusulin / DDPD
  Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]
  ☆84Updated 8 months ago
• ozekri / SEPO
  Code for the paper: "Fine-Tuning Discrete Diffusion Models with Policy Gradient Methods"
  ☆31Updated 7 months ago
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆129Updated 3 weeks ago
• nkalyanv99 / UNI-D2
  ☆53Updated 3 weeks ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆114Updated last year
• zhixuan-lin / forgetting-transformer
  [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning
  ☆134Updated 3 weeks ago
• yilundu / ired_code_release
  ☆82Updated last year
• harryjo97 / RDLM
  Official Code Repository for the paper "Continuous Diffusion Model for Language Modeling" (NeurIPS 2025).
  ☆55Updated 3 months ago
• AllanYangZhou / nfn
  NF-Layers for constructing neural functionals.
  ☆93Updated 2 years ago
• zacharyhorvitz / Fk-Diffusion-Steering
  A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.
  ☆205Updated 6 months ago
• dvruette / gidd
  Code accompanying the paper "Generalized Interpolating Discrete Diffusion"
  ☆112Updated 7 months ago
• microsoft / soc-fine-tuning-sd
  Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching
  ☆59Updated 7 months ago
• igul222 / plaid
  ☆111Updated 2 years ago
• ChenWu98 / algorithmic-creativity
  [ICML 2025] Roll the dice & look before you leap: Going beyond the creative limits of next-token prediction
  ☆82Updated 7 months ago
• Cranial-XIX / longhorn
  Official PyTorch Implementation of the Longhorn Deep State Space Model
  ☆57Updated last year
• HKUNLP / diffusion-vs-ar
  [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"
  ☆87Updated 11 months ago
• kvfrans / splus
  ☆122Updated 7 months ago
• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆48Updated 2 years ago
• MinkaiXu / Energy-Diffusion-LLM
  Reproduce ICLR2025 Energy-Based Diffusion Language Models for Text Generation
  ☆52Updated 5 months ago
• willisma / diffuse_nnx
  A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…
  ☆127Updated 3 months ago
• lindermanlab / elk
  Scalable and Stable Parallelization of Nonlinear RNNS
  ☆28Updated 2 months ago
• lazaratan / meta-flow-matching
  Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold
  ☆69Updated 10 months ago
• GFNOrg / diffusion-finetuning
  ☆41Updated last year
• lebellig / discrete-fm
  Educational implementation of the Discrete Flow Matching paper
  ☆128Updated last year
• ML-GSAI / RADD
  Official PyTorch implementation for "Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Data" (ICLR…
  ☆77Updated 7 months ago
• GFNOrg / GFlowNet-EM
  Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.
  ☆41Updated last year
• IdoAmos / not-from-scratch
  ☆33Updated last year
• DoronHav / WassersteinFlowMatching
  ☆36Updated this week