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.
☆144Oct 16, 2025Updated 5 months ago
Alternatives and similar repositories for diffuse_nnx
Users that are interested in diffuse_nnx are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Minimal but scalable implementation of large language models in JAX☆35Nov 28, 2025Updated 4 months ago
- ☆37Feb 4, 2026Updated 2 months ago
- ☆16Mar 6, 2026Updated last month
- [ICML 2025] Official implementation of Spherical Diffusion Policy: A SE(3) Equivariant Visuomotor Policy with Spherical Fourier Represent…☆39Jul 8, 2025Updated 9 months ago
- A collection of notes and examples on model deployment☆24Mar 9, 2026Updated last month
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Code for Scalable Offline Model-Based RL with Action chunking☆21Feb 20, 2026Updated last month
- Python package for P2 (Path Planning), a masked diffusion model sampling method for sequence generation (protein, text, etc.).☆23Aug 19, 2025Updated 7 months ago
- Official code for ACT: Empowering Decision Transformer with Dynamic Programming via Advantage Conditioning (AAAI'24)☆17Feb 10, 2024Updated 2 years ago
- A benchmark for evaluating reinforcement learning algorithms that train the policies using imaginary rollouts from LLMs.☆14Nov 4, 2025Updated 5 months ago
- implementation of https://arxiv.org/pdf/2312.09299☆21Jul 3, 2024Updated last year
- code for the paper Offline Prioritized Experience Replay☆12Jun 13, 2023Updated 2 years ago
- ☆11Jan 3, 2024Updated 2 years ago
- ☆11Oct 3, 2022Updated 3 years ago
- Stainless neural networks in JAX☆34Feb 3, 2026Updated 2 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting with the flexibility to host WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Cloudways by DigitalOcean.
- Graph-Assisted Stitching for Offline Hierarchical Reinforcement Learning (ICML 2025)☆17Dec 9, 2025Updated 4 months ago
- ☆95May 31, 2025Updated 10 months ago
- Adjoint-Equivariant Neural Network for Semi-Simple Lie Algebras☆32Sep 20, 2025Updated 6 months ago
- ☆13Apr 25, 2024Updated last year
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆12May 8, 2023Updated 2 years ago
- KANs and MLPs☆12Jun 7, 2024Updated last year
- Nonlinear Model-Predictive Control on Resource-Constrained Microcontrollers☆32Feb 26, 2025Updated last year
- ☆14Jun 22, 2025Updated 9 months ago
- Train to 94% on CIFAR-10 in 4.4 seconds on a single A100☆12Dec 30, 2023Updated 2 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆19Jul 24, 2025Updated 8 months ago
- Shanghai Jiao Tong University 2023-2024, CS3601 Operating System☆20Dec 27, 2023Updated 2 years ago
- Implementation of Model Tensor Planning in JAX, TMLR 2025 & ICLR 2026.☆26Jun 5, 2025Updated 10 months ago
- A collection of various llm pruning implementations, training code for GPUs & TPUs, and evaluation script.☆63Feb 18, 2026Updated last month
- Audiocraft is a library for audio processing and generation with deep learning. It features the state-of-the-art EnCodec audio compressor…☆11Aug 7, 2023Updated 2 years ago
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 2 years ago
- Official implementation of Harnessing Mixed Offline Reinforcement Learning Datasets via Trajectory Reweighting☆16Feb 14, 2024Updated 2 years ago
- LocoFormer - Generalist Locomotion via Long-Context Adaptation☆108Mar 27, 2026Updated 2 weeks ago
- JAX Multi-Agent RL, Neuro-Evolution, and A-Life Library☆14Oct 12, 2025Updated 5 months ago
- NordVPN Threat Protection Pro™ • AdTake your cybersecurity to the next level. Block phishing, malware, trackers, and ads. Lightweight app that works with all browsers.
- ☆30May 4, 2023Updated 2 years ago
- ☆48Feb 23, 2025Updated last year
- Codes accompanying the paper "Offline Reinforcement Learning with Value-Based Episodic Memory" (ICLR 2022 https://arxiv.org/abs/2110.0979…☆15Mar 9, 2022Updated 4 years ago
- ☆46Mar 29, 2026Updated last week
- ☆15Jan 18, 2026Updated 2 months ago
- ☆14Oct 11, 2022Updated 3 years ago
- Bullseye Polytope Clean-Label Poisoning Attack☆15Nov 5, 2020Updated 5 years ago