yuanchenyang / smalldiffusion
Simple and readable code for training and sampling from diffusion models
☆207Updated this week
Related projects ⓘ
Alternatives and complementary repositories for smalldiffusion
- ☆197Updated 4 months ago
- A simple implimentation of Bayesian Flow Networks (BFN)☆239Updated 10 months ago
- Run PyTorch in JAX. 🤝☆200Updated last year
- For optimization algorithm research and development.☆449Updated this week
- Visualizations of the theory behind diffusion models.☆75Updated 7 months ago
- Annotated version of the Mamba paper☆457Updated 8 months ago
- From-scratch diffusion model implemented in PyTorch.☆77Updated 9 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆252Updated 5 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆328Updated last month
- 94% on CIFAR-10 in 2.6 seconds 💨 96% in 27 seconds☆177Updated last week
- ☆195Updated last month
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI☆256Updated last week
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆333Updated 3 weeks ago
- Scalable neural net training via automatic normalization in the modular norm.☆121Updated 3 months ago
- A simple library for scaling up JAX programs☆127Updated 2 weeks ago
- Flow-matching algorithms in JAX☆77Updated 3 months ago
- A tool to analyze and debug neural networks in pytorch. Use a GUI to traverse the computation graph and view the data from many different…☆270Updated 3 weeks ago
- σ-GPT: A New Approach to Autoregressive Models☆59Updated 3 months ago
- Simplified Masked Diffusion Language Model☆207Updated last week
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆615Updated last week
- ☆128Updated this week
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆445Updated 3 weeks ago
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆663Updated 6 months ago
- Train VAE like a boss☆246Updated 3 weeks ago
- A minimal yet resourceful implementation of diffusion models (along with pretrained models + synthetic images for nine datasets)☆257Updated 2 months ago
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆404Updated 8 months ago
- ☆292Updated 4 months ago
- Use Jax functions in Pytorch☆228Updated last year
- Minimal Implementation of a D3PM in pytorch☆183Updated 6 months ago
- A mini-library for training consistency models.☆215Updated 10 months ago