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"
  ☆321Updated 4 months ago
• cfifty / rotation_trick
  ☆102Updated last month
• Lakonik / GMFlow
  [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)
  ☆90Updated this week
• yinboc / dito
  Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"
  ☆114Updated 3 months ago
• limefax / rope-nd
  N-dimensional Rotary Position Embeddings for PyTorch
  ☆53Updated last year
• mmathew23 / improved_edm
  Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"
  ☆93Updated last year
• autonomousvision / gta
  [ICLR'24] GTA: A Geometry-Aware Attention Mechanism for Multi-view Transformers
  ☆129Updated last month
• lucidrains / rectified-flow-pytorch
  Implementation of rectified flow and some of its followup research / improvements in Pytorch
  ☆289Updated 3 weeks ago
• sangyun884 / rfpp
  The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024
  ☆112Updated 7 months ago
• leffff / FlowModels
  The aim of this repository is to test and implement Flow-Matching-based models
  ☆92Updated 4 months ago
• lucidrains / VN-transformer
  A Transformer made of Rotation-equivariant Attention using Vector Neurons
  ☆90Updated last year
• lucidrains / autoregressive-diffusion-pytorch
  Implementation of Autoregressive Diffusion in Pytorch
  ☆381Updated 6 months ago
• zbr17 / OptVQ
  Towards training VQ-VAE models robustly!
  ☆75Updated 4 months ago
• CompVis / discrete-interpolants
  The official implementation of "[MASK] is All You Need"
  ☆119Updated 2 months ago
• VinAIResearch / LFM
  Official PyTorch implementation of the paper: Flow Matching in Latent Space
  ☆269Updated 3 months ago
• TheStageAI / TorchIntegral
  Integral Neural Networks in PyTorch
  ☆125Updated 5 months ago
• YangLing0818 / consistency_flow_matching
  Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"
  ☆212Updated 4 months ago
• RobinWu218 / SimDINO
  [ICML 2025] Official Implementation for SimDINO/SimDINOv2
  ☆128Updated 2 months ago
• lebellig / discrete-fm
  Educational implementation of the Discrete Flow Matching paper
  ☆88Updated 8 months ago
• elyxlz / givt-pytorch
  A partial implementation of Generative Infinite Vocabulary Transformer (GIVT) from Google Deepmind, in PyTorch.
  ☆18Updated last year
• kabachuha / OpenMMDiT
  Open(MM)DiT: An Easy, Fast and Memory-Efficient System for (MM)DiT Training and Inference
  ☆32Updated last year
• goombalab / hydra
  Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"
  ☆132Updated 3 months ago
• ppwwyyxx / RAM-multiprocess-dataloader
  Demystify RAM Usage in Multi-Process Data Loaders
  ☆193Updated 2 years ago
• OliverRensu / FlowAR
  “FlowAR: Scale-wise Autoregressive Image Generation Meets Flow Matching” FlowAR employs a simplest scale design and is compatible with an…
  ☆113Updated 2 weeks ago
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆93Updated last year
• valeoai / Halton-MaskGIT
  [ICLR2025] Halton Scheduler for Masked Generative Image Transformer
  ☆231Updated this week
• lucidrains / titok-pytorch
  Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"
  ☆173Updated 10 months ago
• zelaki / eqvae
  [ICML'25] EQ-VAE: Equivariance Regularized Latent Space for Improved Generative Image Modeling.
  ☆102Updated 2 months ago
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆102Updated 10 months ago
• RobinWu218 / ToST
  [ICLR 2025 Spotlight] Official Implementation for ToST (Token Statistics Transformer)
  ☆85Updated 2 months ago