Implementation of Diffusion Transformers and Rectified Flow in Jax
☆27Jul 9, 2024Updated last year
Alternatives and similar repositories for jax-mmdit
Users that are interested in jax-mmdit are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- DiT (training + flow matching) in Jax☆11Jan 5, 2025Updated last year
- JAX Implementations of Descript Audio Codec and EnCodec☆35Mar 30, 2025Updated last year
- JAX port of FLUX.1 models using flax.nnx☆24Sep 28, 2024Updated last year
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆40Feb 9, 2026Updated 2 months ago
- ☆20Jul 12, 2023Updated 2 years ago
- Wordpress hosting with auto-scaling - Free Trial • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- ☆24Jun 4, 2024Updated last year
- A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.☆72Jul 24, 2022Updated 3 years ago
- Trust Region Preference Approximation: A simple and stable reinforcement learning algorithm for LLM reasoning☆15Jun 28, 2025Updated 9 months ago
- Homework 3 for Berkeley CS 280: our version of the MIT Mini Places challenge☆12Mar 5, 2016Updated 10 years ago
- ☆16May 11, 2025Updated 11 months ago
- ☆16Jul 8, 2024Updated last year
- JAX implementation of the T5 model: Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer☆24Jun 10, 2023Updated 2 years ago
- ☆16Dec 30, 2024Updated last year
- ☆24Dec 16, 2024Updated last year
- Wordpress hosting with auto-scaling - Free Trial • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- ☆62Mar 4, 2022Updated 4 years ago
- ☆18Aug 24, 2024Updated last year
- Blog post☆17Feb 16, 2024Updated 2 years ago
- ComfyUI Custom Nodes☆25Dec 31, 2025Updated 3 months ago
- Code for the note "NF4 Isn't Information Theoretically Optimal (and that's Good)☆21Jun 22, 2023Updated 2 years ago
- ☆18Mar 18, 2026Updated 3 weeks ago
- ☆17Aug 13, 2024Updated last year
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Jan 16, 2023Updated 3 years ago
- Unofficial JAX implementations of deep learning research papers☆162Jun 25, 2022Updated 3 years ago
- Wordpress hosting with auto-scaling - Free Trial • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- ☆23Oct 15, 2024Updated last year
- Soft Mixture of Experts Vision Transformer, addressing MoE limitations as highlighted by Puigcerver et al., 2023.☆16Aug 13, 2023Updated 2 years ago
- Voxtral: Convert Mistral into a end2end SpeechLM. No information bottleneck, preserves prosody, learns interruptions from data. Unlike GP…☆46Mar 7, 2025Updated last year
- A set of Python scripts that makes your experience on TPU better☆56Sep 18, 2025Updated 6 months ago
- An implementation of a Brownian motion using ClojureScript with re-frame and Highcharts☆11Feb 8, 2019Updated 7 years ago
- JAX Synergistic Memory Inspector☆187Jul 16, 2024Updated last year
- Brax + Pufferlib + CARBS for gpu-accelerated robotics RL☆12Jun 12, 2025Updated 10 months ago
- User-interactive anime colorization☆10Oct 8, 2020Updated 5 years ago
- Slightly better tools for random prompt generation in ComfyUI☆11Jun 9, 2025Updated 10 months ago
- Wordpress hosting with auto-scaling - Free Trial • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Neuroevolution is a Competitive Alternative to Reinforcement Learning for Skill Discovery☆17Apr 2, 2026Updated 2 weeks ago
- Flow-matching algorithms in JAX☆117Aug 12, 2024Updated last year
- Repository of the WACV'24 paper "Can CLIP Help Sound Source Localization?"☆34Feb 21, 2025Updated last year
- Accelerate, Optimize performance with streamlined training and serving options with JAX.☆353Updated this week
- A Pytorch Lightning WGAN-gp to generate faces☆11Jan 26, 2021Updated 5 years ago
- ☆10Sep 21, 2024Updated last year
- DINO-based perceptual losses and FDD feature extraction☆26Jan 7, 2026Updated 3 months ago