helblazer811 / diffusion-visualizations
Visualizations of the theory behind diffusion models.
☆172Updated 11 months ago
Alternatives and similar repositories for diffusion-visualizations:
Users that are interested in diffusion-visualizations are comparing it to the libraries listed below
- Flow-matching algorithms in JAX☆86Updated 7 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆270Updated 9 months ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆128Updated last year
- ☆36Updated 3 months ago
- ☆150Updated 7 months ago
- Diffusion models in PyTorch☆99Updated last week
- My take on Flow Matching☆44Updated 2 months ago
- Simple and readable code for training and sampling from diffusion models☆453Updated 2 months ago
- code for "Riemannian Flow Matching on General Geometries".☆210Updated last year
- ☆155Updated last month
- Getting crystal-like representations with harmonic loss☆180Updated last month
- A simple implimentation of Bayesian Flow Networks (BFN)☆240Updated last year
- ☆138Updated last year
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆115Updated last month
- 3D Gaussian Splatting in JAX☆57Updated 10 months ago
- Minimal GPT (~350 lines with a simple task to test it)☆62Updated 3 months ago
- Train VAE like a boss☆270Updated 5 months ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆223Updated last month
- Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"☆43Updated 4 months ago
- ☆25Updated last year
- Annotated Flow Matching paper☆178Updated 6 months ago
- 🧱 Modula software package☆187Updated this week
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆70Updated last month
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆23Updated 2 months ago
- The boundary of neural network trainability is fractal☆197Updated last year
- Graph neural networks in JAX.☆67Updated 9 months ago
- Focused on fast experimentation and simplicity☆70Updated 3 months ago
- Use Jax functions in Pytorch☆239Updated last year
- Modern Fixed Point Systems using Pytorch☆88Updated last year
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆421Updated 3 months ago