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.
☆148Oct 16, 2025Updated 7 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:
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 years ago
- Minimal but scalable implementation of large language models in JAX☆35Nov 28, 2025Updated 5 months ago
- ☆41Feb 4, 2026Updated 3 months ago
- Official implementation of 'A Large-Scale Exploration of mu-Transfer'☆32Jun 5, 2025Updated 11 months ago
- ☆18May 7, 2026Updated last week
- 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.
- [ICML 2025] Official implementation of Spherical Diffusion Policy: A SE(3) Equivariant Visuomotor Policy with Spherical Fourier Represent…☆42Jul 8, 2025Updated 10 months ago
- A collection of notes and examples on model deployment☆24Mar 9, 2026Updated 2 months ago
- Code for Scalable Offline Model-Based RL with Action chunking☆26Feb 20, 2026Updated 3 months ago
- Collaborative retina modelling across datasets and species.☆20Apr 10, 2026Updated last month
- GRUD is a genotype imputation based on deep learning algorithms☆12Jul 3, 2024Updated last year
- ☆33Mar 1, 2023Updated 3 years ago
- Official code for ACT: Empowering Decision Transformer with Dynamic Programming via Advantage Conditioning (AAAI'24)☆17Feb 10, 2024Updated 2 years ago
- SDK For J60 Joint Module Control☆17Aug 20, 2025Updated 9 months ago
- A benchmark for evaluating reinforcement learning algorithms that train the policies using imaginary rollouts from LLMs.☆14Nov 4, 2025Updated 6 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.
- code for the paper Offline Prioritized Experience Replay☆12Jun 13, 2023Updated 2 years ago
- ☆11Oct 3, 2022Updated 3 years ago
- 4-bit Shampoo for Memory-Efficient Network Training (NeurIPS 2024)☆13Feb 13, 2025Updated last year
- [NeurIPS 2025] Better Tokens for Better 3D: Advancing Vision-Language Modeling in 3D Medical Imaging☆40Nov 4, 2025Updated 6 months ago
- Papers on Topology + Learning.☆14Feb 12, 2020Updated 6 years ago
- ☆13Apr 25, 2024Updated 2 years ago
- KANs and MLPs☆12Jun 7, 2024Updated last year
- ☆14Jun 22, 2025Updated 10 months ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆19Jul 24, 2025Updated 9 months ago
- AI Agents on 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.
- Shanghai Jiao Tong University 2023-2024, CS3601 Operating System☆20Dec 27, 2023Updated 2 years ago
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 2 years ago
- Harmonization of multi-site, multi-shell diffusion MRI☆14Oct 16, 2025Updated 7 months ago
- ☆49Feb 23, 2025Updated last year
- Algorithm implementation for our paper: Robotic World Model: A Neural Network Simulator for Robust Policy Optimization in Robotics☆53Jan 26, 2026Updated 3 months ago
- 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
- ☆16Apr 14, 2026Updated last month
- Tools and Utils for Experiments (TUX)☆16Jan 28, 2025Updated last year
- ☆21Sep 6, 2021Updated 4 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- [AAAI'2024] Critic-Guided Decision Transformer for Offline Reinforcement Learning☆18May 21, 2025Updated last year
- ☆16Mar 2, 2022Updated 4 years ago
- Super Mario Bros. (NES) gameplay dataset for machine learning.☆13Jul 22, 2025Updated 9 months ago
- JAX Scalify: end-to-end scaled arithmetics☆18Oct 30, 2024Updated last year
- ☆22Jan 23, 2024Updated 2 years ago
- Implementation of various equivariant models in JAX☆19Apr 12, 2024Updated 2 years ago
- [NeurIPS 2023] Refining Diffusion Planner for Reliable Behavior Synthesis by Automatic Detection of Infeasible Plans☆22Jan 31, 2024Updated 2 years ago