ivan-chai / torch-linear-assignmentLinks
Batch computation of the linear assignment problem on GPU.
☆92Updated last month
Alternatives and similar repositories for torch-linear-assignment
Users that are interested in torch-linear-assignment are comparing it to the libraries listed below
Sorting:
- Pytorch implementation for MeanFlow☆119Updated last month
- [ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"☆387Updated 8 months ago
- Pytorch implementation of MeanFlow on ImageNet and CIFAR10☆220Updated last week
- N-dimensional Rotary Position Embeddings for PyTorch☆68Updated last year
- [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)☆128Updated 3 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆403Updated 10 months ago
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆149Updated 7 months ago
- JAX implementation of MeanFlow☆344Updated last month
- ☆143Updated 5 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆370Updated 3 weeks ago
- FlexTok: Resampling Images into 1D Token Sequences of Flexible Length☆248Updated 3 months ago
- [ICML 2025] Official Implementation for SimDINO/SimDINOv2☆165Updated 5 months ago
- The official implementation of "[MASK] is All You Need"☆124Updated last month
- ☆144Updated 2 weeks ago
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆178Updated last year
- Towards training VQ-VAE models robustly!☆83Updated last month
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆237Updated 7 months ago
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆601Updated 9 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆422Updated 7 months ago
- Demystify RAM Usage in Multi-Process Data Loaders☆197Updated 2 years ago
- [ICLR2025] Halton Scheduler for Masked Generative Image Transformer☆256Updated 2 months ago
- ☆298Updated 10 months ago
- Fast Multi-dimensional Sparse Attention☆602Updated 2 weeks ago
- This repository includes the official implementation of our paper "Beyond Next-Token: Next-X Prediction for Autoregressive Visual Generat…☆233Updated 4 months ago
- PyTorch re-implementation for MeanFlow☆95Updated last month
- DDT: Decoupled Diffusion Transformer☆273Updated last week
- ☆269Updated 8 months ago
- ☆254Updated 3 months ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆104Updated last week
- Code for Fast Training of Diffusion Models with Masked Transformers☆408Updated last year