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"
  ☆435Updated 3 months ago
• Lakonik / GMFlow
  [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)
  ☆177Updated 2 months ago
• Gsunshine / py-meanflow
  Pytorch implementation for MeanFlow
  ☆307Updated 6 months ago
• limefax / rope-nd
  N-dimensional Rotary Position Embeddings for PyTorch
  ☆79Updated last year
• lucidrains / autoregressive-diffusion-pytorch
  Implementation of Autoregressive Diffusion in Pytorch
  ☆431Updated 2 months ago
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆126Updated 5 months ago
• cfifty / rotation_trick
  ☆160Updated 10 months ago
• lucidrains / rectified-flow-pytorch
  Implementation of rectified flow and some of its followup research / improvements in Pytorch
  ☆421Updated last week
• lucidrains / titok-pytorch
  Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"
  ☆183Updated last year
• elyxlz / givt-pytorch
  A partial implementation of Generative Infinite Vocabulary Transformer (GIVT) from Google Deepmind, in PyTorch.
  ☆21Updated last year
• Neur-IO / OptVQ
  Towards training VQ-VAE models robustly!
  ☆91Updated 6 months ago
• Jhomanik / Optimal-Flow-Matching
  The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)
  ☆109Updated last year
• lucidrains / ema-pytorch
  A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
  ☆637Updated last month
• ppwwyyxx / RAM-multiprocess-dataloader
  Demystify RAM Usage in Multi-Process Data Loaders
  ☆205Updated 2 years ago
• lucidrains / VN-transformer
  A Transformer made of Rotation-equivariant Attention using Vector Neurons
  ☆101Updated 2 years ago
• kyleliang919 / C-Optim
  When it comes to optimizers, it's always better to be safe than sorry
  ☆402Updated 4 months ago
• lucidrains / mmdit
  Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch
  ☆510Updated 2 weeks ago
• CompVis / discrete-interpolants
  The official implementation of "[MASK] is All You Need"
  ☆126Updated 6 months ago
• yinboc / dito
  Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"
  ☆164Updated last year
• YangLing0818 / consistency_flow_matching
  Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"
  ☆257Updated last year
• lucidrains / adam-atan2-pytorch
  Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch
  ☆134Updated 3 months ago
• HaoyiZhu / MeanFlow-PyTorch
  PyTorch re-implementation for MeanFlow
  ☆116Updated 6 months ago
• lucidrains / rotary-embedding-torch
  Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch
  ☆796Updated 6 months ago
• sangyun884 / rfpp
  The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024
  ☆127Updated last year
• zhuyu-cs / MeanFlow
  Pytorch implementation of MeanFlow on ImageNet and CIFAR10
  ☆414Updated 3 weeks ago
• apple / ml-tarflow
  ☆320Updated last year
• VinAIResearch / LFM
  Official PyTorch implementation of the paper: Flow Matching in Latent Space
  ☆331Updated last year
• mmathew23 / improved_edm
  Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"
  ☆98Updated last year
• locuslab / ect
  Consistency Models Made Easy
  ☆324Updated last year
• ArchiMickey / Just-a-DiT
  A repo of a modified version of Diffusion Transformer
  ☆45Updated 4 months ago