A simple and intuitive library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with simplified mathematical explanations. Made as part of my journey for learning and experimenting with generative AI.
☆40Jul 27, 2026Updated last month
Alternatives and similar repositories for FlaxDiff
Users that are interested in FlaxDiff are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Diffusion Probabilistic Model in Jax☆13Apr 20, 2024Updated 2 years ago
- Implementation of numerous Vision Transformers in Google's JAX and Flax.☆22Aug 30, 2022Updated 4 years ago
- Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist☆66Dec 16, 2023Updated 2 years ago
- Official repository for the Monte Carlo guided Diffusion for Bayesian linear inverse problems paper:☆21Dec 16, 2024Updated last year
- A python module that performas a convolution with a kernel with a variable width☆14Mar 4, 2016Updated 10 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- A simple one file Python project to run 2D phase field problems with linear quadrilaterals☆21Aug 25, 2025Updated last year
- GPU-accelerated differentiable sparse linear solver library for JAX☆48Updated this week
- Some useful tools for mesh processing☆12Sep 9, 2025Updated 11 months ago
- Solving Inverse Problems with Diffusion Optimal Control [NeurIPS 2024]☆19Dec 21, 2024Updated last year
- A python code for 2d topology optimization using MMA optimizers in NLOPT☆12Feb 9, 2017Updated 9 years ago
- Large-Area Fabrication-Aware Computational Diffractive Optics (SIGGRAPH Asia & TOG 2025)☆17Nov 20, 2025Updated 9 months ago
- Minimal multi-gpu implementation of EDM2: "Analyzing and Improving the Training Dynamics of Diffusion Models"☆41Mar 5, 2024Updated 2 years ago
- Generative models with JAX.☆22Feb 21, 2026Updated 6 months ago
- Clean single-file implementation of offline RL algorithms in JAX☆184Jun 5, 2026Updated 2 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Implementation of the ByteDance MagicMix paper☆19Nov 4, 2022Updated 3 years ago
- ⛔️ DEPRECATED GPU Ocean Python/CUDA codebase☆11Nov 9, 2023Updated 2 years ago
- Consistency Conditions for any two X-ray images.☆14Jun 14, 2026Updated 2 months ago
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆267Mar 21, 2025Updated last year
- Optimal Transport for Machine Learners☆85Updated this week
- PyTorch Implementation of DSB for Score Based Generative Modeling. Experiments managed using Hydra.☆210Nov 23, 2021Updated 4 years ago
- Chirp instantaneous frequency estimation using stochastic differential equation Gaussian processes☆13Oct 30, 2024Updated last year
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆39Mar 18, 2025Updated last year
- Examples on how to use Fortran and C++ mex files with Matlab☆12Jul 7, 2017Updated 9 years ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- ☆43Jun 29, 2025Updated last year
- Getting Started in Imitation Learning☆13Mar 3, 2025Updated last year
- ☆18Aug 6, 2024Updated 2 years ago
- Matlab codes for hotspot analysis to avoid overheating in Topology optimization for Additive Manufacturing☆12Dec 12, 2025Updated 8 months ago
- PrideDiff: Physics-Regularized Generalized Diffusion Model for CT Reconstruction☆12Dec 10, 2024Updated last year
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆33Mar 21, 2024Updated 2 years ago
- SParse lINear solvers in JAX☆53Aug 4, 2026Updated 3 weeks ago
- Official repository for the article Compositional Discrete Latent Code for High Fidelity, Productive Diffusion Models (https://arxiv.org/…☆38Sep 5, 2025Updated 11 months ago
- Python package based on PyTorch and Dask for efficient and parallelized data assimilation of weather model data and observations on GPUs.☆16Nov 18, 2025Updated 9 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Average-Reward Reinforcement Learning with Trust Region Methods☆11Oct 17, 2022Updated 3 years ago
- share NLIGA code which fork form sourceforge☆16Aug 18, 2020Updated 6 years ago
- A Declarative Language for Expressing Partial World Knowledge to Reinforcement Learning Agents☆17Jan 19, 2024Updated 2 years ago
- GPFlow based implementation of temporal parallellization of state space GPs☆16Sep 8, 2022Updated 3 years ago
- PINNs with irreversible constraints.☆17Jun 10, 2026Updated 2 months ago
- [AAAI'25 Oral] Are Expressive Models Truly Necessary for Offline RL?☆15Dec 10, 2024Updated last year
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆34May 12, 2023Updated 3 years ago