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 3 weeks ago
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
- 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
- ☆15May 11, 2025Updated last year
- 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.
- Implementation of the ByteDance MagicMix paper☆19Nov 4, 2022Updated 3 years ago
- Implementation of numerous Vision Transformers in Google's JAX and Flax.☆22Aug 30, 2022Updated 3 years ago
- Minimal (truly) muP implementation, consistent with TP4 and TP5 papers notation☆14Jan 2, 2026Updated 4 months ago
- Gradient-informed particle MCMC methods☆12Jan 29, 2024Updated 2 years ago
- JAX Implementations of Descript Audio Codec and EnCodec☆36Mar 30, 2025Updated last year
- Chirp instantaneous frequency estimation using stochastic differential equation Gaussian processes☆13Oct 30, 2024Updated last year
- ☆15Sep 4, 2025Updated 8 months ago
- 🚀 Official code for “XStreamVGGT: Extremely Memory-Efficient Streaming Vision Geometry Grounded Transformer with KV Cache Compression”, …☆45Jan 27, 2026Updated 3 months ago
- Average-Reward Reinforcement Learning with Trust Region Methods☆11Oct 17, 2022Updated 3 years 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.
- Stochastic trace estimation using JAX☆17Aug 20, 2025Updated 9 months ago
- Curated list of JAX Resources and Packages☆40Apr 30, 2026Updated 2 weeks ago
- ☆26Dec 19, 2025Updated 5 months ago
- Autonomous ML research infrastructure for autoresearch by Karpathy. Multi-GPU parallelism, structured experiment tracking, adaptive searc…☆94Apr 18, 2026Updated last month
- [NeurIPS 2025] LabUtopia: High-Fidelity Simulation and Hierarchical Benchmark for Scientific Embodied Agents☆37Apr 15, 2026Updated last month
- Implementation of Computer Vision Models in JAX (equinox)☆23May 7, 2026Updated last week
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 years ago
- GPFlow based implementation of temporal parallellization of state space GPs☆16Sep 8, 2022Updated 3 years ago
- Fast semantic search for biorXiv manuscripts☆12Feb 16, 2025Updated last year
- 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.
- The open source code of ALMTokenizer2: Towards Low bit-rate and Semantic-rich Audio Tokenizer with Flow-based Scalar Diffusion Transforme…☆45Sep 5, 2025Updated 8 months ago
- USD models for NVIDIA Isaac Sim☆18Aug 8, 2025Updated 9 months ago
- User-interactive anime colorization☆10Oct 8, 2020Updated 5 years ago
- Speech-To-Text forced-alignment Speech processing Universal PERformance Benchmark☆36May 7, 2025Updated last year
- A language server for bibfile citations☆16Jan 8, 2026Updated 4 months ago
- Unofficial Implementation of Null-text Inversion (https://arxiv.org/abs/2211.09794)☆12Nov 20, 2022Updated 3 years ago
- Copy top-level annotations from one URL (and/or group) to another☆14Apr 29, 2021Updated 5 years ago
- General Purpose Point Cloud Feature Extractor☆13Mar 1, 2018Updated 8 years ago
- A Declarative Language for Expressing Partial World Knowledge to Reinforcement Learning Agents☆17Jan 19, 2024Updated 2 years ago
- 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.
- （WIP）long form speech generatoins☆31Apr 2, 2025Updated last year
- novoStoic2.0: Integrated Pathway Design Tool with Thermodynamic Considerations and Enzyme Selection☆11Apr 28, 2025Updated last year
- The companion code for the paper "Variational inference via Wasserstein gradient flows (W-VI) M. Lambert, S. Chewi, F. Bach, S. Bonnabel…☆17Feb 1, 2023Updated 3 years ago
- Unofficial Pytorch Lightning Implementation of "Real-time Speech Frequency Bandwidth Extension"☆41Oct 20, 2025Updated 7 months ago
- faster inference☆28Jan 20, 2025Updated last year
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆40Apr 14, 2026Updated last month
- A tiny reinforcement learning codebase for continuous control, built on top of JAX.☆15Mar 28, 2023Updated 3 years ago