ivan-chai / torch-linear-assignmentLinks
Batch computation of the linear assignment problem on GPU.
☆96Updated 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:
- [ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"☆408Updated last week
- Pytorch implementation for MeanFlow☆223Updated 3 months ago
- [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)☆144Updated 3 weeks ago
- N-dimensional Rotary Position Embeddings for PyTorch☆76Updated last year
- ☆151Updated 7 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆416Updated last year
- Towards training VQ-VAE models robustly!☆85Updated 3 months ago
- PyTorch re-implementation for MeanFlow☆105Updated 3 months ago
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆181Updated last year
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆157Updated 9 months ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆112Updated 2 months ago
- ☆165Updated last week
- Demystify RAM Usage in Multi-Process Data Loaders☆204Updated 2 years ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆389Updated 2 months ago
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆124Updated last year
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆465Updated 9 months ago
- Pytorch implementation of MeanFlow on ImageNet and CIFAR10☆331Updated 2 months ago
- JAX implementation of MeanFlow☆464Updated 3 months ago
- FlexTok: Resampling Images into 1D Token Sequences of Flexible Length☆263Updated 5 months ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆134Updated 3 weeks ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆246Updated 9 months ago
- When it comes to optimizers, it's always better to be safe than sorry☆376Updated last month
- [ICML 2025] Official Implementation for SimDINO/SimDINOv2☆180Updated 7 months ago
- The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)☆95Updated 10 months ago
- A Transformer made of Rotation-equivariant Attention using Vector Neurons☆93Updated 2 years ago
- The official implementation of "[MASK] is All You Need"☆125Updated 3 months ago
- ☆292Updated 10 months ago
- This repository includes the official implementation of our paper "Beyond Next-Token: Next-X Prediction for Autoregressive Visual Generat…☆238Updated 3 weeks ago
- A partial implementation of Generative Infinite Vocabulary Transformer (GIVT) from Google Deepmind, in PyTorch.☆20Updated last year
- “FlowAR: Scale-wise Autoregressive Image Generation Meets Flow Matching” FlowAR employs a simplest scale design and is compatible with an…☆157Updated 6 months ago