helblazer811 / diffusion-visualizations
Visualizations of the theory behind diffusion models.
☆79Updated 10 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☆85Updated 6 months ago
- NF-Layers for constructing neural functionals.☆82Updated last year
- My take on Flow Matching☆38Updated last month
- Artificial Kuramoto Oscillatory Neurons☆61Updated 2 weeks ago
- 3D Gaussian Splatting in JAX☆56Updated 9 months ago
- Neural Optimal Transport with Lagrangian Costs☆53Updated 7 months ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆24Updated 8 months ago
- Code for the paper: Rotating Features for Object Discovery☆50Updated 6 months ago
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆45Updated 2 years ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆61Updated this week
- A Tutorial for Diffusion Models☆45Updated last year
- ☆56Updated 8 months ago
- ☆22Updated 2 months ago
- ☆40Updated 3 months ago
- ☆49Updated last year
- ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC☆137Updated 4 months ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆92Updated 2 weeks ago
- Annotated Flow Matching paper☆173Updated 5 months ago
- Modern Fixed Point Systems using Pytorch☆86Updated last year
- Use Jax functions in Pytorch☆237Updated last year
- Diffusion models in PyTorch☆92Updated last week
- All about the fundamentals and working of Diffusion Models☆153Updated 2 years ago
- WIP☆93Updated 6 months ago
- Graph neural networks in JAX.☆67Updated 8 months ago
- Cellular Automata Accelerated in JAX (Oral at ICLR 2025).☆81Updated last week
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆17Updated last year
- [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"☆52Updated 10 months ago
- ☆161Updated 3 months ago
- A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!☆71Updated last month