A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its variants as the primary backbone with support for ImageNet training and various sampling strategies.
☆137Oct 16, 2025Updated 4 months ago
Alternatives and similar repositories for diffuse_nnx
Users that are interested in diffuse_nnx are comparing it to the libraries listed below
Sorting:
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 years ago
- ☆16Oct 20, 2025Updated 4 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Jun 5, 2025Updated 8 months ago
- ☆30May 4, 2023Updated 2 years ago
- [IROS 2025] D4orm: Multi-Robot Trajectories with Dynamics-aware Diffusion Denoised Deformations.☆48Sep 4, 2025Updated 5 months ago
- Stainless neural networks in JAX☆34Feb 3, 2026Updated 3 weeks ago
- Intriguing Properties of Data Attribution on Diffusion Models (ICLR 2024)☆37Jan 23, 2024Updated 2 years ago
- [IROS2025] & [RA-L2025] Official Code for GRaD-Nav series☆59Dec 8, 2025Updated 2 months ago
- ☆33Mar 1, 2023Updated 3 years ago
- This repo contains code for detecting poultry barns from high-resolution aerial imagery and an accompanying dataset of predicted barns ov…☆38Oct 11, 2022Updated 3 years ago
- ☆270Updated this week
- Graph-Assisted Stitching for Offline Hierarchical Reinforcement Learning (ICML 2025)☆17Dec 9, 2025Updated 2 months ago
- Two-stream remote sensing model for water quality mapping: 2SeaColor☆10Feb 2, 2021Updated 5 years ago
- Framework for geospatial research CNN development.☆11Apr 10, 2021Updated 4 years ago
- A fine-mapping method integrating GWAS summary statistics and functional annotation data☆11Dec 28, 2023Updated 2 years ago
- Parrot robot Jumping Sumo☆10Jan 8, 2016Updated 10 years ago
- A gui for evaluating estimated trajctory☆10Oct 7, 2021Updated 4 years ago
- ☆43Dec 1, 2025Updated 3 months ago
- SR-DiT Speedrunning ImageNet Diffusion☆125Dec 31, 2025Updated 2 months ago
- ☆17Dec 30, 2025Updated 2 months ago
- JAX Scalify: end-to-end scaled arithmetics☆18Oct 30, 2024Updated last year
- Linking field-based and airborne data☆11Apr 5, 2020Updated 5 years ago
- [ICLR 2022] Boosting Randomized Smoothing with Variance Reduced Classifiers☆12Mar 29, 2022Updated 3 years ago
- ☆11Jan 3, 2024Updated 2 years ago
- Framework for Algorithmic Correctness Testing of Operators☆16Feb 20, 2026Updated last week
- Perception-centric Pedagogical Framework For Autonomous Aerial Robots☆13Apr 2, 2025Updated 10 months ago
- 🚀 Collection of libraries used with fms-hf-tuning to accelerate fine-tuning and training of large models.☆13Jan 30, 2026Updated last month
- Implementation of various equivariant models in JAX☆12Apr 12, 2024Updated last year
- ☆12Mar 25, 2025Updated 11 months ago
- GORILLA: Guiding-center ORbit Integration with Local Linearization Approach☆15Updated this week
- Weekly data science workshops for SIG AIDA at UIUC.☆12Jun 22, 2022Updated 3 years ago
- Autonomous robot exploration in unknown outdoor environment☆10Nov 13, 2018Updated 7 years ago
- ☆13Jun 22, 2025Updated 8 months ago
- ☆10Jun 26, 2023Updated 2 years ago
- Repository for various academic and professional CV versions☆10Jan 16, 2026Updated last month
- Teaching a Convolutional Neural Network to recognize painting genre. Handcrafted dataset. Cool visualizations.☆10Dec 19, 2018Updated 7 years ago
- A repo collating papers and other material related to neural radiance fields (NeRFs), neural scene representations and associated works w…☆13Jul 24, 2023Updated 2 years ago
- A Python package designed for research in diffusion-based generative modeling☆31Nov 26, 2025Updated 3 months ago
- ☆11Oct 15, 2023Updated 2 years ago