Minimal JAX implementation unifying Diffusion and Flow Matching algorithms as alternative strategies for transporting data distributions.
☆64Dec 19, 2025Updated 4 months ago
Alternatives and similar repositories for jax_generative_models
Users that are interested in jax_generative_models are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Generative Modeling via Drifting in MLX☆42Feb 6, 2026Updated 2 months ago
- Corax: Core RL in JAX☆41Feb 22, 2024Updated 2 years ago
- Deep learning chat based on DL4J☆11Mar 31, 2017Updated 9 years ago
- Definitive implementation of the stochastic interpolant framework for generative modeling in jax.☆40Jul 13, 2025Updated 9 months ago
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆38Mar 18, 2025Updated last year
- 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.
- UMAP in pure MLX for Apple Silicon. 30x faster than umap-learn.☆42Mar 5, 2026Updated last month
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Aug 8, 2022Updated 3 years ago
- Zephyr is a declarative neural network library on top of JAX allowing for easy and fast neural network designing, creation, and manipulat…☆39Mar 21, 2026Updated 3 weeks ago
- Euclidean Wasserstein-2 optimal transportation☆48Aug 19, 2023Updated 2 years ago
- ☆18Mar 18, 2026Updated last month
- Japanese Mahjong (Riichi Mahjong) Font with OpenType|OpenType 機能付き麻雀牌図フォント☆21Mar 17, 2026Updated last month
- Sampling and visualising the von Mises-Fisher distribution. An important hyper-spherical distribution.☆15Jun 5, 2020Updated 5 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆57May 21, 2023Updated 2 years ago
- Map (deep learning) model weights between different model implementations.☆19Mar 9, 2026Updated last month
- 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.
- ☆18Jul 10, 2022Updated 3 years ago
- ACT-Bench – We Evaluate Action-Fidelity of World Models for Autonomous Driving☆28Dec 23, 2024Updated last year
- ☆14Oct 19, 2019Updated 6 years ago
- Drop-in environment replacements that make your RL algorithm train faster.☆22Jun 19, 2024Updated last year
- Pessimistic Value Iteration for Multi-Task Data Sharing in Offline RL☆18Nov 21, 2023Updated 2 years ago
- A fast, local neural text to speech system☆18Feb 24, 2025Updated last year
- Unofficial Implementation of Null-text Inversion (https://arxiv.org/abs/2211.09794)☆12Nov 20, 2022Updated 3 years ago
- ☆10Sep 21, 2024Updated last year
- ☆10Dec 18, 2023Updated 2 years ago
- 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.
- Joint trajectory planning for constrained manipulation using the Closed-Chain Affordance framework by Janak Panthi☆13Mar 19, 2026Updated last month
- ☆23Jan 15, 2026Updated 3 months ago
- Getting Started in Imitation Learning☆13Mar 3, 2025Updated last year
- Code for Posterior Sampling for Deep Reinforcement Learning, ICML 2023☆28Mar 7, 2024Updated 2 years ago
- ☆17Mar 16, 2022Updated 4 years ago
- JAX exponential map normalising flows on sphere☆17Oct 4, 2020Updated 5 years ago
- A microframework for creating command-line applications in Zig☆48Updated this week
- Official implementation for the paper: "Real-Time Inverse Kinematics for Generating Multi-Constrained Movements of Virtual Human Characte…☆21Feb 19, 2026Updated 2 months ago
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆18Nov 24, 2025Updated 4 months 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.
- Reading Group @mila-iqia on Computational Optimal Transport for Machine Learning Applications☆13Jun 3, 2022Updated 3 years ago
- Official Implementation of SFM and the baselines in Jax.☆21May 31, 2025Updated 10 months ago
- Zero-Cost Whole-Body Teleoperation for Mobile Manipulation☆12Mar 4, 2025Updated last year
- Concept Learning Dynamics☆16Oct 29, 2024Updated last year
- Official implementation of NeurIPS'23 paper, Beyond Uniform Sampling: Offline Reinforcement Learning with Imbalanced Datasets☆26Jan 29, 2024Updated 2 years ago
- Reinforcement learning pipeline for specific configured robot to learn to reach randomly sampled target poses within its workspace.☆22Aug 28, 2025Updated 7 months ago
- Pointax: PointMaze Environment for JAX☆27Oct 22, 2025Updated 5 months ago