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
  ☆132Updated 7 months ago
• yilundu / ired_code_release
  ☆73Updated last year
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆102Updated 2 months ago
• ChenWu98 / algorithmic-creativity
  [ICML 2025] Roll the dice & look before you leap: Going beyond the creative limits of next-token prediction
  ☆71Updated 4 months ago
• Cranial-XIX / longhorn
  Official PyTorch Implementation of the Longhorn Deep State Space Model
  ☆55Updated 10 months ago
• AllanYangZhou / nfn
  NF-Layers for constructing neural functionals.
  ☆90Updated last year
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆101Updated last week
• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆48Updated 2 years ago
• kvfrans / splus
  ☆120Updated 4 months ago
• riverstone496 / awesome-second-order-optimization
  ☆28Updated 2 weeks ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆105Updated last year
• microsoft / ReinMax
  Beyond Straight-Through
  ☆102Updated 2 years ago
• zhixuan-lin / forgetting-transformer
  [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning
  ☆131Updated 3 weeks ago
• harryjo97 / RDLM
  Official Code Repository for the paper "Continuous Diffusion Model for Language Modeling" (NeurIPS 2025).
  ☆44Updated 3 weeks ago
• liusulin / DDPD
  Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]
  ☆79Updated 5 months ago
• ozekri / SEPO
  Code for the paper: "Fine-Tuning Discrete Diffusion Models with Policy Gradient Methods"
  ☆27Updated 4 months ago
• radarFudan / mamba-minimal-jax
  ☆33Updated 10 months ago
• IdoAmos / not-from-scratch
  ☆33Updated 11 months ago
• HKUNLP / diffusion-vs-ar
  [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"
  ☆77Updated 8 months ago
• dvruette / gidd
  Code accompanying the paper "Generalized Interpolating Discrete Diffusion"
  ☆103Updated 4 months ago
• lucidrains / scaling-vin-pytorch
  Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group
  ☆37Updated last year
• watcl-lab / positional_attention
  Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"
  ☆14Updated 4 months ago
• shawntan / stickbreaking-attention
  Stick-breaking attention
  ☆60Updated 3 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
• bm371613 / conditional-flow-matching
  ☆14Updated last year
• SHI-Yu-Zhe / generative-modeling-explained
  Ying Nian Wu's UCLA Statistical Machine Learning Tutorial on generative modeling.
  ☆61Updated 2 years ago
• tml-epfl / why-weight-decay
  Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]
  ☆68Updated last year
• igul222 / plaid
  ☆107Updated 2 years ago
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆58Updated 4 months ago
• varun-ml / diffusion-models-tutorial
  Experiment with diffusion models that you can run on your local jupyter instances
  ☆63Updated 11 months ago