Minimal JAX implementation unifying Diffusion and Flow Matching algorithms as alternative strategies for transporting data distributions.
☆68Dec 19, 2025Updated 10 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:
- streaming deep reinforcement learning but 4x faster with jax!☆19Jan 4, 2026Updated 10 months ago
- ☆16Sep 4, 2025Updated last year
- A Python package designed for research in diffusion-based generative modeling☆32Aug 16, 2026Updated 2 months ago
- ☆130Feb 25, 2025Updated last year
- A collection of algorithms and experiment tools for safe sim to real transfer in robotics.☆30May 19, 2026Updated 5 months ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Clean single-file implementation of offline RL algorithms in JAX☆184Jun 5, 2026Updated 5 months ago
- Flow-matching algorithms in JAX☆119Aug 12, 2024Updated 2 years ago
- An Agent-Environment interface with Jax transformation compatibility☆13Nov 17, 2023Updated 2 years ago
- [ICML 2025] Local Manifold Approximation and Projection for Manifold-Aware Diffusion Planning☆20Jun 5, 2025Updated last year
- Generative Modeling via Drifting in MLX☆44Feb 6, 2026Updated 9 months ago
- Model predictive path integral control in jax☆19Feb 8, 2026Updated 8 months ago
- Distilled version of D4orm: Multi-robot trajectory smoother☆14Jun 3, 2026Updated 5 months ago
- This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."☆40Jul 11, 2025Updated last year
- Actor-Free Continuous Control via Structurally Maximizable Q-Functions☆14Dec 15, 2025Updated 10 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Differentiable linear model predictive control in Python☆15Dec 10, 2025Updated 10 months ago
- A JAX library for Sequential Convex Programming (SCP) Model Predictive Control (MPC) path planning☆37Jan 22, 2024Updated 2 years ago
- The official implementation of "Horizon Reduction Makes RL Scalable"☆210Aug 2, 2025Updated last year
- Real-time motor control for seamless development, testing, and deployment across simulation and hardware☆10Oct 18, 2026Updated 2 weeks ago
- Towards better athletic intelligence☆32Jan 31, 2026Updated 9 months ago
- Model Predictive Parkour Control of a Monoped Hopper in Dynamically Changing Environments☆25May 20, 2025Updated last year
- A minimal, single-file implementation of the MeanFlow paper on 2D toy examples, with a side-by-side comparison to rectified flow.☆27Jul 4, 2025Updated last year
- GPU-friendly, auto-differentiable LQR solver with JAX.☆226Aug 8, 2025Updated last year
- Implementation of Model Tensor Planning in JAX, TMLR 2025 & ICLR 2026.☆47May 20, 2026Updated 5 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- RL and MARL with Flax NNX☆16Sep 7, 2026Updated 2 months ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆142Dec 20, 2023Updated 2 years ago
- ☆29Jan 8, 2026Updated 9 months ago
- Generative Modeling with Flux Matching☆71May 11, 2026Updated 5 months ago
- Sampling-based model predictive control on GPU with JAX/MJX☆318Jul 18, 2026Updated 3 months ago
- Pointax: PointMaze Environment for JAX☆28Oct 22, 2025Updated last year
- [NeurIPS 2023] Implementation of Elastic Decision Transformer☆41Oct 12, 2023Updated 3 years ago
- Custom triton kernels for training Karpathy's nanoGPT.☆19Oct 21, 2024Updated 2 years ago
- Drop-in environment replacements that make your RL algorithm train faster.☆23Jun 19, 2024Updated 2 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Scaling safe exploration to vision control☆15Feb 19, 2025Updated last year
- off-policy RL on long sequences☆173May 29, 2026Updated 5 months ago
- Official implementation for "How Should We Meta-Learn Reinforcement Learning Algorithms?"☆23Sep 7, 2025Updated last year
- Supplementary Data for "A graph representation of molecular ensembles for polymer property prediction"☆25Oct 12, 2022Updated 4 years ago
- Official implementation of Stackelberg PPO for morphology –control co-design.☆19Mar 17, 2026Updated 7 months ago
- Challenging reinforcement learning environments with locomotion tasks in space☆14Jun 27, 2022Updated 4 years ago
- Algorithm implementation for our paper: Robotic World Model: A Neural Network Simulator for Robust Policy Optimization in Robotics☆59Jan 26, 2026Updated 9 months ago