Alternatives and similar repositories for modified-shortcut-models-pytorch

Users that are interested in modified-shortcut-models-pytorch are comparing it to the libraries listed below

• Jhomanik / Optimal-Flow-Matching
  The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)
  ☆93Updated 9 months ago
• apple / ml-tarflow
  ☆281Updated 9 months ago
• milmor / diffusion-transformer
  Implementation of Diffusion Transformer Model in Pytorch
  ☆70Updated 5 months ago
• jadehaus / preference-flow-matching
  Official code implementation for the work Preference Alignment with Flow Matching (NeurIPS 2024)
  ☆58Updated 11 months ago
• kylehkhsu / latent_quantization
  ☆41Updated last year
• lebellig / flow-matching
  Annotated Flow Matching paper
  ☆209Updated last year
• sangyun884 / rfpp
  The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024
  ☆123Updated 11 months ago
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆78Updated 9 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
  ☆66Updated 2 months ago
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆98Updated this week
• raywang4 / EqM
  ☆34Updated last week
• CompVis / tread
  ☆128Updated last month
• YangLing0818 / consistency_flow_matching
  Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"
  ☆243Updated 8 months ago
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆105Updated last year
• yinboc / dito
  Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"
  ☆151Updated 8 months ago
• apple / ml-agm
  ☆44Updated last year
• sangyun884 / fast-ode
  Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023
  ☆88Updated 7 months ago
• AaronHavens / FlowMatching
  A repo where I play with conditional flow approaches for learning time-varying vector-fields.
  ☆23Updated last year
• Lakonik / GMFlow
  [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)
  ☆138Updated 4 months ago
• lucidrains / scaling-vin-pytorch
  Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group
  ☆37Updated last year
• lucidrains / vit-arc-slot
  Explorations into improving ViTArc with Slot Attention
  ☆43Updated 11 months ago
• lucidrains / rectified-flow-pytorch
  Implementation of rectified flow and some of its followup research / improvements in Pytorch
  ☆381Updated 2 months ago
• LabForComputationalVision / memorization_generalization_in_diffusion_models
  ☆49Updated last year
• giannisdaras / ambient-tweedie
  [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"
  ☆53Updated last year
• gstoica27 / DeltaFM
  [ICCV 2025] Official Implementation of Contrastive Flow Matching
  ☆125Updated 3 months ago
• m1balcerak / EnergyMatching
  Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling" [NeurIPS 2025]
  ☆152Updated this week
• xandergos / sCM-mnist
  Unofficial implementation of "Simplifying, Stabilizing & Scaling Continuous-Time Consistency Models" for MNIST
  ☆77Updated 6 months ago
• leffff / FlowModels
  The aim of this repository is to test and implement Flow-Matching-based models
  ☆113Updated 8 months ago
• ZehaoDou-official / FPS-SMC-2023
  Official Pytorch repository for "Diffusion Posterior Sampling for Linear Inverse Problem Solving: A Filtering Perspective", where FPS (Fi…
  ☆47Updated last year
• lucidrains / improving-transformers-world-model-for-rl
  Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch
  ☆136Updated 5 months ago