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"
  ☆327Updated 5 months ago
• Lakonik / GMFlow
  [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)
  ☆100Updated last week
• yinboc / dito
  Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"
  ☆121Updated 4 months ago
• lucidrains / VN-transformer
  A Transformer made of Rotation-equivariant Attention using Vector Neurons
  ☆92Updated last year
• autonomousvision / gta
  [ICLR'24] GTA: A Geometry-Aware Attention Mechanism for Multi-view Transformers
  ☆129Updated last month
• limefax / rope-nd
  N-dimensional Rotary Position Embeddings for PyTorch
  ☆54Updated last year
• lucidrains / autoregressive-diffusion-pytorch
  Implementation of Autoregressive Diffusion in Pytorch
  ☆382Updated 7 months ago
• andrehuang / loftup
  Official Code for "LoftUp: Learning a Coordinate-Based Feature Upsampler for Vision Foundation Models"
  ☆150Updated 3 weeks ago
• RobinWu218 / SimDINO
  [ICML 2025] Official Implementation for SimDINO/SimDINOv2
  ☆139Updated 2 months ago
• cfifty / rotation_trick
  ☆112Updated 2 months ago
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆96Updated last year
• lucidrains / rectified-flow-pytorch
  Implementation of rectified flow and some of its followup research / improvements in Pytorch
  ☆294Updated last week
• lucidrains / hl-gauss-pytorch
  The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch
  ☆63Updated this week
• lucidrains / titok-pytorch
  Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"
  ☆173Updated 11 months ago
• zbr17 / OptVQ
  Towards training VQ-VAE models robustly!
  ☆74Updated 4 months ago
• goombalab / hydra
  Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"
  ☆137Updated 4 months ago
• apple / ml-tarflow
  ☆191Updated 5 months ago
• lucidrains / mmdit
  Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
  ☆361Updated 4 months ago
• lucidrains / adam-atan2-pytorch
  Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch
  ☆106Updated 6 months ago
• CompVis / discrete-interpolants
  The official implementation of "[MASK] is All You Need"
  ☆121Updated 3 months ago
• kyleliang919 / C-Optim
  When it comes to optimizers, it's always better to be safe than sorry
  ☆237Updated 2 months ago
• mmathew23 / improved_edm
  Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"
  ☆94Updated last year
• OliverRensu / xAR
  This repository includes the official implementation of our paper "Beyond Next-Token: Next-X Prediction for Autoregressive Visual Generat…
  ☆209Updated last month
• valeoai / Halton-MaskGIT
  [ICLR2025] Halton Scheduler for Masked Generative Image Transformer
  ☆238Updated 3 weeks ago
• YangLing0818 / consistency_flow_matching
  Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"
  ☆216Updated 4 months ago
• hustvl / DiG
  [CVPR 2025] DiG: Scalable and Efficient Diffusion Models with Gated Linear Attention
  ☆166Updated 3 months ago
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆167Updated 2 months ago
• lucidrains / hyper-connections
  Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public
  ☆83Updated 3 months ago
• sangyun884 / rfpp
  The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024
  ☆115Updated 7 months ago
• MCG-NJU / DDT
  DDT: Decoupled Diffusion Transformer
  ☆252Updated 2 weeks ago