changgyhub / nanoDiffusionLinks
The simplest diffusion model in PyTorch, with Apple M chip acceleration support.
☆39Updated last year
Alternatives and similar repositories for nanoDiffusion
Users that are interested in nanoDiffusion are comparing it to the libraries listed below
Sorting:
- My take on Flow Matching☆84Updated 10 months ago
- From-scratch diffusion model implemented in PyTorch.☆98Updated last year
- Implementation of the proposed Spline-Based Transformer from Disney Research☆105Updated last year
- A minimal implementation of a denoising diffusion model in PyTorch.☆130Updated last year
- Train VAE like a boss☆300Updated last year
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆181Updated last year
- Implementation of Diffusion Transformer Model in Pytorch☆71Updated 6 months ago
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆123Updated 7 months ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆123Updated last year
- Reproduction of DDPO paper (RLHF for diffusion)☆93Updated 2 years ago
- Implementation of a multimodal diffusion transformer in Pytorch☆106Updated last year
- Implementation of a framework for Genie2 in Pytorch☆153Updated 10 months ago
- Just another reasonably minimal repo for class-conditional training of pixel-space diffusion transformers.☆136Updated 5 months ago
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆103Updated last month
- Implementation of a transformer for reinforcement learning using `x-transformers`☆68Updated last month
- Annotated Flow Matching paper☆221Updated last year
- Implement a MNIST(also minimal) version of denoising diffusion probabilistic model from scratch.The model only has 4.55MB.☆134Updated 2 years ago
- Focused on fast experimentation and simplicity☆75Updated 10 months ago
- PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlight☆149Updated 2 years ago
- Text to Image Latent Diffusion using a Transformer core☆214Updated last year
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆57Updated last year
- 🦾 EvalGIM (pronounced as "EvalGym") is an evaluation library for generative image models. It enables easy-to-use, reproducible automatic…☆88Updated 11 months ago
- WIP☆93Updated last year
- N-dimensional Rotary Position Embeddings for PyTorch☆77Updated last year
- [ICML 2025] Roll the dice & look before you leap: Going beyond the creative limits of next-token prediction☆75Updated 5 months ago
- Experiment with diffusion models that you can run on your local jupyter instances☆63Updated last year
- Mixture-of-Transformers: A Sparse and Scalable Architecture for Multi-Modal Foundation Models. TMLR 2025.☆122Updated 2 months ago
- Official code implementation for the work Preference Alignment with Flow Matching (NeurIPS 2024)☆62Updated last year
- Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"☆344Updated 3 years ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Updated last year