lucidrains / maskbit-pytorch
Implementation of the proposed MaskBit from Bytedance AI
☆58Updated 2 weeks ago
Related projects ⓘ
Alternatives and complementary repositories for maskbit-pytorch
- Implementation of a multimodal diffusion transformer in Pytorch☆97Updated 4 months ago
- ☆101Updated 4 months ago
- Scaling Diffusion Transformers with Mixture of Experts☆202Updated 2 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆181Updated last week
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆161Updated 4 months ago
- The official PyTorch implementation for Improving Long-Text Alignment for Text-to-Image Diffusion Models (LongAlign)☆54Updated 3 weeks ago
- Explorations into improving ViTArc with Slot Attention☆36Updated 3 weeks ago
- Official PyTorch Implementation of "Scalable Autoregressive Image Generation with Mamba"☆109Updated 2 months ago
- Implementation of a single layer of the MMDiT, proposed in Stable Diffusion 3, in Pytorch☆253Updated 2 months ago
- ☆26Updated 6 months ago
- A big_vision inspired repo that implements a generic Auto-Encoder class capable in representation learning and generative modeling.☆29Updated 4 months ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆94Updated 2 weeks ago
- The codebase of our paper "Improving the Training of Rectified Flows"☆80Updated 3 weeks ago
- Implementation of a Light Recurrent Unit in Pytorch☆46Updated last month
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆108Updated 3 weeks ago
- ☆23Updated last month
- Rectified Diffusion: Straightness Is Not Your Need☆118Updated 2 weeks ago
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆53Updated 5 months ago
- Implementation of Agent Attention in Pytorch☆86Updated 4 months ago
- Implementation of Infini-Transformer in Pytorch☆104Updated last month
- [NeurIPS 2024] Learning-to-Cache: Accelerating Diffusion Transformer via Layer Caching☆71Updated 3 months ago
- Official PyTorch Implementation of "FlowTurbo: Towards Real-time Flow-Based Image Generation with Velocity Refiner"☆60Updated last month
- Implementation of the proposed Spline-Based Transformer from Disney Research☆74Updated this week
- Implementation of a framework for Gamengen in Pytorch☆90Updated last month
- ☆26Updated last week
- ☆78Updated 10 months ago
- [NeurIPS 2024] Stabilize the Latent Space for Image Autoregressive Modeling: A Unified Perspective☆38Updated last week
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆117Updated 3 months ago
- Open(MM)DiT: An Easy, Fast and Memory-Efficient System for (MM)DiT Training and Inference☆21Updated 7 months ago