A simple, easy-to-understand 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.
☆41Apr 29, 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:
- Implementation of Diffusion Transformers and Rectified Flow in Jax☆27Jul 9, 2024Updated last year
- Diffusion Probabilistic Model in Jax☆13Apr 20, 2024Updated 2 years ago
- JAX bindings for Flash Attention v2☆106Feb 28, 2026Updated 3 months ago
- Code repository for Particle Denoising Diffusion Sampler☆15Apr 2, 2024Updated 2 years ago
- Official Pytorch repository for "Diffusion Posterior Sampling for Linear Inverse Problem Solving: A Filtering Perspective", where FPS (Fi…☆50Aug 6, 2024Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- This repository contains the Julia code for the paper "Competitive Gradient Descent"☆25Dec 18, 2019Updated 6 years ago
- [Science Advances] Data-driven Global Ocean Modeling for Seasonal to Decadal Prediction☆23Sep 9, 2025Updated 9 months ago
- [ACL 2025] OZSpeech: One-step Zero-shot Speech Synthesis with Learned-Prior-Conditioned Flow Matching☆45Feb 9, 2025Updated last year
- ☆15May 11, 2025Updated last year
- Benchmark evaluating ocean forecasting systems against reference datasets and observations.☆41Updated this week
- Solving Inverse Problems with Diffusion Optimal Control [NeurIPS 2024]☆19Dec 21, 2024Updated last year
- Implementation of the ByteDance MagicMix paper☆19Nov 4, 2022Updated 3 years ago
- Next generation MCMC samplers with automatic differentiaion and adaptive Poisson thinning☆13Apr 20, 2026Updated last month
- Implementation of numerous Vision Transformers in Google's JAX and Flax.☆22Aug 30, 2022Updated 3 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Gradient-informed particle MCMC methods☆12Jan 29, 2024Updated 2 years ago
- JAX Implementations of Descript Audio Codec and EnCodec☆37Mar 30, 2025Updated last year
- Chirp instantaneous frequency estimation using stochastic differential equation Gaussian processes☆13Oct 30, 2024Updated last year
- Regional Ocean Forecasting with Hierarchical Graph Neural Networks☆21Aug 7, 2025Updated 10 months ago
- ☆15Sep 4, 2025Updated 9 months ago
- 🚀 Official code for “XStreamVGGT: Extremely Memory-Efficient Streaming Vision Geometry Grounded Transformer with KV Cache Compression”, …☆45Jan 27, 2026Updated 4 months ago
- run tinygrad kernels on esp32☆14Nov 28, 2023Updated 2 years ago
- Incremental Disentanglement for Environment-Aware Zero-Shot Text-to-Speech Synthesis☆27Mar 21, 2025Updated last year
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆161Nov 1, 2022Updated 3 years 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.
- [NeurIPS 2025] LabUtopia: High-Fidelity Simulation and Hierarchical Benchmark for Scientific Embodied Agents☆36May 29, 2026Updated last week
- Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist☆67Dec 16, 2023Updated 2 years ago
- GPFlow based implementation of temporal parallellization of state space GPs☆16Sep 8, 2022Updated 3 years ago
- The open source code of ALMTokenizer2: Towards Low bit-rate and Semantic-rich Audio Tokenizer with Flow-based Scalar Diffusion Transforme…☆45Sep 5, 2025Updated 9 months ago
- MuChin: A Chinese Colloquial Description Benchmark for Evaluating Language Models in the Field of Music☆26Jan 7, 2026Updated 5 months ago
- Official implementation of Categorical Flow Maps on text.☆58Feb 16, 2026Updated 3 months ago
- Speech-To-Text forced-alignment Speech processing Universal PERformance Benchmark☆38May 7, 2025Updated last year
- Neuroevolution is a Competitive Alternative to Reinforcement Learning for Skill Discovery☆17Apr 2, 2026Updated 2 months ago
- USD models for NVIDIA Isaac Sim☆19Aug 8, 2025Updated 10 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Unofficial Implementation of Null-text Inversion (https://arxiv.org/abs/2211.09794)☆12Nov 20, 2022Updated 3 years ago
- TABR-BERT: an Accurate and Robust BERT-based Transfer Learning Model for TCR-pMHC Interaction Prediction☆12Jul 19, 2024Updated last year
- Unofficial Pytorch Lightning Implementation of "Real-time Speech Frequency Bandwidth Extension"☆41Oct 20, 2025Updated 7 months ago
- Code repository for paper: A score-based filter for nonlinear data assimilation☆14Nov 5, 2024Updated last year
- A Pytorch Lightning WGAN-gp to generate faces☆11Jan 26, 2021Updated 5 years ago
- ☆10Sep 21, 2024Updated last year
- faster inference☆28Jan 20, 2025Updated last year