ivan-chai / torch-linear-assignmentLinks
Batch computation of the linear assignment problem on GPU.
☆83Updated 5 months ago
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:
- [ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"☆338Updated 6 months ago
- JAX implementation of MeanFlow☆148Updated this week
- ☆117Updated 2 months ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆97Updated last year
- N-dimensional Rotary Position Embeddings for PyTorch☆56Updated last year
- [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)☆105Updated last month
- Implementation of Autoregressive Diffusion in Pytorch☆392Updated 7 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆314Updated last week
- Demystify RAM Usage in Multi-Process Data Loaders☆194Updated 2 years ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆108Updated 7 months ago
- ☆205Updated 6 months ago
- Towards training VQ-VAE models robustly!☆74Updated 5 months ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆64Updated 3 weeks ago
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆95Updated last year
- [ICLR2025] Halton Scheduler for Masked Generative Image Transformer☆239Updated last month
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆114Updated 8 months ago
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆174Updated 3 months ago
- [ICML 2025] Official Implementation for SimDINO/SimDINOv2☆143Updated 3 months ago
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆125Updated 4 months ago
- When it comes to optimizers, it's always better to be safe than sorry☆244Updated 2 months ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆221Updated 5 months ago
- The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)☆77Updated 6 months ago
- ☆131Updated last year
- A Pytorch Implementation of Finite Scalar Quantization☆138Updated last year
- The aim of this repository is to test and implement Flow-Matching-based models☆96Updated 5 months ago
- Educational implementation of the Discrete Flow Matching paper☆92Updated 10 months ago
- The official implementation of "[MASK] is All You Need"☆121Updated 4 months ago
- Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public☆85Updated last week
- “FlowAR: Scale-wise Autoregressive Image Generation Meets Flow Matching” FlowAR employs a simplest scale design and is compatible with an…☆127Updated last month
- This repository includes the official implementation of our paper "Beyond Next-Token: Next-X Prediction for Autoregressive Visual Generat…☆215Updated last month