Alternatives and similar repositories for torch-linear-assignment

Users that are interested in torch-linear-assignment are comparing it to the libraries listed below

• naver-ai / rope-vit
  [ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"
  ☆399Updated 9 months ago
• Gsunshine / py-meanflow
  Pytorch implementation for MeanFlow
  ☆145Updated last month
• limefax / rope-nd
  N-dimensional Rotary Position Embeddings for PyTorch
  ☆72Updated last year
• Lakonik / GMFlow
  [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)
  ☆140Updated 3 months ago
• lucidrains / autoregressive-diffusion-pytorch
  Implementation of Autoregressive Diffusion in Pytorch
  ☆408Updated 10 months ago
• cfifty / rotation_trick
  ☆151Updated 5 months ago
• zhuyu-cs / MeanFlow
  Pytorch implementation of MeanFlow on ImageNet and CIFAR10
  ☆249Updated last month
• Neur-IO / OptVQ
  Towards training VQ-VAE models robustly!
  ☆83Updated 2 months ago
• lucidrains / rectified-flow-pytorch
  Implementation of rectified flow and some of its followup research / improvements in Pytorch
  ☆376Updated last month
• yinboc / dito
  Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"
  ☆151Updated 7 months ago
• HaoyiZhu / MeanFlow-PyTorch
  PyTorch re-implementation for MeanFlow
  ☆96Updated 2 months ago
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆108Updated last month
• lucidrains / titok-pytorch
  Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"
  ☆180Updated last year
• ppwwyyxx / RAM-multiprocess-dataloader
  Demystify RAM Usage in Multi-Process Data Loaders
  ☆201Updated 2 years ago
• CompVis / discrete-interpolants
  The official implementation of "[MASK] is All You Need"
  ☆124Updated 2 months ago
• Gsunshine / meanflow
  JAX implementation of MeanFlow
  ☆368Updated last month
• elyxlz / givt-pytorch
  A partial implementation of Generative Infinite Vocabulary Transformer (GIVT) from Google Deepmind, in PyTorch.
  ☆20Updated last year
• leffff / FlowModels
  The aim of this repository is to test and implement Flow-Matching-based models
  ☆106Updated 8 months ago
• lucidrains / mmdit
  Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
  ☆446Updated 8 months ago
• apple / ml-flextok
  FlexTok: Resampling Images into 1D Token Sequences of Flexible Length
  ☆253Updated 3 months ago
• YangLing0818 / consistency_flow_matching
  Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"
  ☆240Updated 8 months ago
• Hhhhhhao / continuous_tokenizer
  ☆265Updated 3 months ago
• sangyun884 / rfpp
  The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024
  ☆122Updated 11 months ago
• VinAIResearch / LFM
  Official PyTorch implementation of the paper: Flow Matching in Latent Space
  ☆299Updated 8 months ago
• OliverRensu / FlowAR
  “FlowAR: Scale-wise Autoregressive Image Generation Meets Flow Matching” FlowAR employs a simplest scale design and is compatible with an…
  ☆153Updated 4 months ago
• OliverRensu / xAR
  This repository includes the official implementation of our paper "Beyond Next-Token: Next-X Prediction for Autoregressive Visual Generat…
  ☆234Updated 4 months ago
• valeoai / Halton-MaskGIT
  [ICLR2025] Halton Scheduler for Masked Generative Image Transformer
  ☆260Updated 2 months ago
• hustvl / DiG
  [CVPR 2025] DiG: Scalable and Efficient Diffusion Models with Gated Linear Attention
  ☆172Updated 6 months ago
• MCG-NJU / DDT
  DDT: Decoupled Diffusion Transformer
  ☆275Updated last month
• SHI-Labs / NATTEN
  Fast Multi-dimensional Sparse Attention
  ☆621Updated last month