helblazer811 / diffusion-visualizations
Visualizations of the theory behind diffusion models.
☆75Updated 7 months ago
Related projects ⓘ
Alternatives and complementary repositories for diffusion-visualizations
- Flow-matching algorithms in JAX☆77Updated 3 months ago
- ☆48Updated 9 months ago
- Neural Optimal Transport with Lagrangian Costs☆49Updated 4 months ago
- 3D Gaussian Splatting in JAX☆54Updated 5 months ago
- ☆46Updated 5 months ago
- Run PyTorch in JAX. 🤝☆200Updated last year
- Graph neural networks in JAX.☆67Updated 5 months ago
- A Tutorial for Diffusion Models☆39Updated last year
- Code for the paper: Rotating Features for Object Discovery☆48Updated 3 months ago
- ☆100Updated this week
- Implementation of the proposed Spline-Based Transformer from Disney Research☆76Updated last week
- ☆40Updated 4 months ago
- NF-Layers for constructing neural functionals.☆75Updated 10 months ago
- My take on Flow Matching☆24Updated last week
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆84Updated last month
- A simple implimentation of Bayesian Flow Networks (BFN)☆239Updated 10 months ago
- Neural Diffusion Processes☆73Updated 3 months ago
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆43Updated last year
- Use Jax functions in Pytorch☆228Updated last year
- simple bibtex generator for any text with \cite{}☆31Updated 4 months ago
- code for "Riemannian Flow Matching on General Geometries".☆177Updated 8 months ago
- Cellular Automata Accelerated in JAX☆70Updated 2 weeks ago
- ☆195Updated last month
- Modern Fixed Point Systems using Pytorch☆82Updated last year
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆123Updated 11 months ago
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆21Updated 4 months ago
- Deep Networks Grok All the Time and Here is Why☆17Updated 6 months ago
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆16Updated 2 months ago
- Simple and readable code for training and sampling from diffusion models☆207Updated this week
- The boundary of neural network trainability is fractal☆161Updated 9 months ago