Alternatives and similar repositories for discreteVAE

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

• cfifty / rotation_trick
  ☆159Updated 9 months ago
• jadehaus / preference-flow-matching
  Official code implementation for the work Preference Alignment with Flow Matching (NeurIPS 2024)
  ☆65Updated last year
• minyoungg / vqtorch
  ☆144Updated last year
• lucidrains / h-net-dynamic-chunking
  Implementation of the dynamic chunking mechanism in H-net by Hwang et al. of Carnegie Mellon
  ☆66Updated 2 weeks ago
• erogol / BlaGPT
  Experimental playground for benchmarking language model (LM) architectures, layers, and tricks on smaller datasets. Designed for flexible…
  ☆95Updated this week
• lebellig / discrete-fm
  Educational implementation of the Discrete Flow Matching paper
  ☆129Updated last year
• RobinKa / discrete-flow-matching-pytorch
  Discrete Flow Matching implemented in PyTorch
  ☆34Updated 10 months ago
• duchenzhuang / FSQ-pytorch
  A Pytorch Implementation of Finite Scalar Quantization
  ☆173Updated 2 years ago
• lucidrains / self-reasoning-tokens-pytorch
  Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto
  ☆57Updated last year
• Jhomanik / Optimal-Flow-Matching
  The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)
  ☆101Updated last year
• kylehkhsu / latent_quantization
  ☆44Updated 2 years ago
• lebellig / flow-matching
  Annotated Flow Matching paper
  ☆226Updated last year
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆107Updated last year
• lucidrains / adam-atan2-pytorch
  Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch
  ☆134Updated 3 months ago
• sony / sqvae
  Pytorch implementation of stochastically quantized variational autoencoder (SQ-VAE)
  ☆193Updated 3 years ago
• lucidrains / rvq-vae-gpt
  My attempts at applying Soundstream design on learned tokenization of text and then applying hierarchical attention to text generation
  ☆90Updated last year
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆126Updated 5 months ago
• google-deepmind / md4
  Official Jax Implementation of MD4 Masked Diffusion Models
  ☆152Updated 11 months ago
• zhixuan-lin / forgetting-transformer
  [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning
  ☆137Updated last month
• apple / ml-tarflow
  ☆318Updated last year
• lucidrains / hyper-connections
  Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public
  ☆163Updated last week
• igul222 / plaid
  ☆111Updated 2 years ago
• fal-ai-community / nano-mdm
  Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun
  ☆56Updated 10 months ago
• YangLing0818 / consistency_flow_matching
  Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"
  ☆257Updated last year
• ozekri / SEPO
  Code for the paper: "Fine-Tuning Discrete Diffusion Models with Policy Gradient Methods"
  ☆32Updated 8 months ago
• yukara-ikemiya / modified-shortcut-models-pytorch
  PyTorch implementation of Shortcut Models [Frans, 2025] with little modification
  ☆70Updated 6 months ago
• ML-GSAI / RADD
  Official PyTorch implementation for "Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Data" (ICLR…
  ☆78Updated 7 months ago
• thjashin / multires-conv
  Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)
  ☆127Updated 2 years ago
• Lakonik / GMFlow
  [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)
  ☆175Updated 2 months ago
• goombalab / hydra
  Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"
  ☆170Updated 11 months ago