Minimal JAX implementation unifying Diffusion and Flow Matching algorithms as alternative strategies for transporting data distributions.
☆66Dec 19, 2025Updated 6 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☆43Feb 6, 2026Updated 5 months ago
- Corax: Core RL in JAX☆42Feb 22, 2024Updated 2 years ago
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆39Mar 18, 2025Updated last year
- Zephyr is a declarative neural network library on top of JAX allowing for easy and fast neural network designing, creation, and manipulat…☆40Mar 21, 2026Updated 3 months ago
- Fast Laplacian estimation☆16May 31, 2024Updated 2 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Optimal Transport for Machine Learners☆78Updated this week
- Author's implementation of ReBRAC, a minimalist improvement upon TD3+BC☆19Oct 22, 2023Updated 2 years ago
- A Python library for variational inference with normalizing flow and annealing☆15Mar 13, 2025Updated last year
- ☆18Jul 10, 2022Updated 3 years ago
- Drop-in environment replacements that make your RL algorithm train faster.☆22Jun 19, 2024Updated 2 years ago
- Neuroevolution is a Competitive Alternative to Reinforcement Learning for Skill Discovery☆17Apr 2, 2026Updated 3 months ago
- Unofficial Implementation of Null-text Inversion (https://arxiv.org/abs/2211.09794)☆12Nov 20, 2022Updated 3 years ago
- ☆10Sep 21, 2024Updated last year
- Joint trajectory planning for constrained manipulation using the Closed-Chain Affordance framework by Janak Panthi☆14May 23, 2026Updated last month
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- ☆18Jul 22, 2025Updated 11 months ago
- ☆23May 22, 2026Updated last month
- Flash attention implementation Minimal CUDA implementation of Flash Attention with tiled computation and online softmax. Educational imp…☆21Dec 27, 2025Updated 6 months ago
- Getting Started in Imitation Learning☆13Mar 3, 2025Updated last year
- 技育CAMP ハッカソンvol.5 開発を効率化するアプリケーションを作ろう！ 最優秀賞 & 技育展2021 開発/スキル支援 部門 最優秀賞☆28Mar 6, 2023Updated 3 years ago
- JAX exponential map normalising flows on sphere☆17Oct 4, 2020Updated 5 years ago
- Official implementation for the paper: "Real-Time Inverse Kinematics for Generating Multi-Constrained Movements of Virtual Human Characte…☆20Feb 19, 2026Updated 4 months ago
- Reading Group @mila-iqia on Computational Optimal Transport for Machine Learning Applications☆13Jun 3, 2022Updated 4 years ago
- Official Implementation of SFM and the baselines in Jax.☆21May 31, 2025Updated last year
- 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.
- Zero-Cost Whole-Body Teleoperation for Mobile Manipulation☆12Mar 4, 2025Updated last year
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆25Dec 29, 2023Updated 2 years ago
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆29Jan 14, 2025Updated last year
- Code base for NeurIPS 2022 paper Curriculum Reinforcement Learning using Optimal Transport via Gradual Domain Adaptation.☆11Aug 21, 2023Updated 2 years ago
- Pointax: PointMaze Environment for JAX☆28Oct 22, 2025Updated 8 months ago
- ☆30Oct 24, 2025Updated 8 months ago
- IntLLaMA: A fast and light quantization solution for LLaMA☆19Jul 21, 2023Updated 2 years ago
- EXTD :: Extremely Tiny Face Detector via Iterative Filter Reuse☆13Jul 25, 2019Updated 6 years ago
- This library has the basic types to avoid use `undefined` or `null`.☆12Jun 15, 2024Updated 2 years 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.
- Clean single-file implementation of offline RL algorithms in JAX☆182Jun 5, 2026Updated last month
- Tutorial kit for building a 3D deep reinforcement learning environment with Unity ML-Agents.☆11Oct 22, 2021Updated 4 years ago
- A generator for golang alfred workflow that helps you create boilerplate code.☆13Jun 26, 2024Updated 2 years ago
- Hyperbolic Busemann Learning with Ideal Prototypes, NeurIPS2021☆26Dec 9, 2021Updated 4 years ago
- Kernel Stein Discrepancy Descent : a method to sample from unnormalized densities☆22Apr 13, 2024Updated 2 years ago
- A collection of awesome projects using MuJoCo.☆17May 27, 2025Updated last year
- This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."☆39Jul 11, 2025Updated 11 months ago