Minimal JAX implementation unifying Diffusion and Flow Matching algorithms as alternative strategies for transporting data distributions.
☆66Dec 19, 2025Updated 5 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:
- 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 10 months ago
- Corax: Core RL in JAX☆41Feb 22, 2024Updated 2 years ago
- Definitive implementation of the stochastic interpolant framework for generative modeling in jax.☆41Jul 13, 2025Updated 10 months ago
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆39Mar 18, 2025Updated last year
- UMAP in pure MLX for Apple Silicon. 30x faster than umap-learn.☆43Mar 5, 2026Updated 2 months ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Minimal Decision Transformer Implementation written in Jax (Flax).☆18Aug 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 2 months ago
- Euclidean Wasserstein-2 optimal transportation☆48Aug 19, 2023Updated 2 years ago
- Fast Laplacian estimation☆16May 31, 2024Updated last year
- ☆18Apr 17, 2026Updated last month
- Sampling and visualising the von Mises-Fisher distribution. An important hyper-spherical distribution.☆16Jun 5, 2020Updated 5 years ago
- ☆13Feb 4, 2025Updated last year
- ☆18Jul 10, 2022Updated 3 years ago
- ACT-Bench – We Evaluate Action-Fidelity of World Models for Autonomous Driving☆29Dec 23, 2024Updated last year
- 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.
- Drop-in environment replacements that make your RL algorithm train faster.☆22Jun 19, 2024Updated last year
- Unofficial Implementation of Null-text Inversion (https://arxiv.org/abs/2211.09794)☆12Nov 20, 2022Updated 3 years ago
- A Pytorch Lightning WGAN-gp to generate faces☆11Jan 26, 2021Updated 5 years ago
- ☆10Sep 21, 2024Updated last year
- ☆10Dec 18, 2023Updated 2 years ago
- PyTorch Implementation of Deep Feature Consistent Variational Autoencoder.☆12Jun 21, 2022Updated 3 years ago
- 先進ゲノム支援中級者講習会2019☆10Oct 11, 2019Updated 6 years ago
- GyroSPD: Vector-valued Distance and Gyrocalculus on the Space of Symmetric Positive Definite Matrices☆17Nov 8, 2021Updated 4 years ago
- Joint trajectory planning for constrained manipulation using the Closed-Chain Affordance framework by Janak Panthi☆13Updated this week
- 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.
- ☆17Jul 22, 2025Updated 10 months ago
- ☆23Jan 15, 2026Updated 4 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
- 技育CAMP ハッカソンvol.5 開発を効率化するアプリケーションを作ろう！ 最優秀賞 & 技育展2021 開発/スキル支援 部門 最優秀賞☆28Mar 6, 2023Updated 3 years ago
- Official implementation for the paper: "Real-Time Inverse Kinematics for Generating Multi-Constrained Movements of Virtual Human Characte…☆20Feb 19, 2026Updated 3 months ago
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆18Nov 24, 2025Updated 6 months ago
- A microframework for creating command-line applications in Zig☆49Apr 17, 2026Updated last month
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- MCTS-RNA is a computational tool for solving RNA inverse folding problem with controlling the GC-content of the RNA sequence very precise…☆13Apr 3, 2017Updated 9 years ago
- 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 11 months ago
- Zero-Cost Whole-Body Teleoperation for Mobile Manipulation☆12Mar 4, 2025Updated last year
- Concept Learning Dynamics☆16Oct 29, 2024Updated last year
- A recreation of the Lorentz embeddings paper from ICML 2018 (Nickel 2018)☆21Jan 16, 2019Updated 7 years ago
- code for the paper Imitation Learning from Observation with Automatic Discount Scheduling☆13Mar 27, 2024Updated 2 years ago