limefax / rope-nd
N-dimensional Rotary Position Embeddings for PyTorch
☆40Updated last year
Alternatives and similar repositories for rope-nd:
Users that are interested in rope-nd are comparing it to the libraries listed below
- Official Pytorch implementation for LARP: Tokenizing Videos with a Learned Autoregressive Generative Prior (ICLR 2025 Oral).☆51Updated 2 weeks ago
- [CVPR 2025] DiG: Scalable and Efficient Diffusion Models with Gated Linear Attention☆124Updated this week
- “FlowAR: Scale-wise Autoregressive Image Generation Meets Flow Matching” FlowAR employs a simplest scale design and is compatible with an…☆90Updated 2 months ago
- Scalable Diffusion Models with State Space Backbone☆151Updated 11 months ago
- Minimal multi-gpu implementation of EDM2: "Analyzing and Improving the Training Dynamics of Diffusion Models"☆28Updated 11 months ago
- ☆75Updated 4 months ago
- The official implementation of "[MASK] is All You Need"☆111Updated this week
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆97Updated 4 months ago
- [CVPR 2025] CoDe: Collaborative Decoding Makes Visual Auto-Regressive Modeling Efficient☆79Updated this week
- [CVPR 2025] Open implementation of "RandAR"☆57Updated last month
- A PyTorch implementation of the paper "Revisiting Non-Autoregressive Transformers for Efficient Image Synthesis"☆42Updated 8 months ago
- The official implementation for "MonoFormer: One Transformer for Both Diffusion and Autoregression"☆84Updated 4 months ago
- The official PyTorch implementation of Fast Diffusion Model☆95Updated last year
- "SlimFlow: Training Smaller One-Step Diffusion Models with Rectified Flow", Yuanzhi Zhu, Xingchao Liu, Qiang Liu☆45Updated 3 months ago
- ☆43Updated 6 months ago
- [NeurIPS 24] Alleviating Distortion in Image Generation via Multi-Resolution Diffusion Models☆36Updated 5 months ago
- Towards training VQ-VAE models robustly!☆54Updated last month
- ☆68Updated 3 months ago
- [ICLR 2025] Rectified Diffusion: Straightness Is Not Your Need☆189Updated 2 months ago
- Adaptive Length Image Tokenization via Recurrent Allocation | How many tokens is an image worth ?☆107Updated 3 weeks ago
- Unofficial implementation of "SODA: Bottleneck Diffusion Models for Representation Learning"☆82Updated 11 months ago
- 🔥stable, simple, state-of-the-art VQVAE toolkit & cookbook☆82Updated 8 months ago
- [NeurIPS 2024] Stabilize the Latent Space for Image Autoregressive Modeling: A Unified Perspective☆62Updated 4 months ago
- This is the official implementation for ControlVAR.☆95Updated 2 months ago
- Inference-only implementation of "One-Step Diffusion Distillation through Score Implicit Matching" [NIPS 2024]☆78Updated 3 months ago