antonioterpin / jkonet-starView external linksLinks
JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".
☆37Mar 18, 2025Updated 10 months ago
Alternatives and similar repositories for jkonet-star
Users that are interested in jkonet-star are comparing it to the libraries listed below
Sorting:
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆42May 8, 2022Updated 3 years ago
- Implicit Differentiable Optimal Control (IDOC) with JAX☆12May 11, 2022Updated 3 years ago
- ☆20Dec 7, 2025Updated 2 months ago
- Fast Laplacian estimation☆16May 31, 2024Updated last year
- [ICCV 2025 Oral] RePoseD: Efficient Relative Pose Estimation With Known Depth Information☆28Jan 12, 2026Updated last month
- 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 7 months ago
- Code release for "Stochastic Optimal Control Matching"☆39Aug 14, 2024Updated last year
- Minimal JAX implementation unifying Diffusion and Flow Matching algorithms as alternative strategies for transporting data distributions.☆62Dec 19, 2025Updated last month
- Optimal Control Tutorials☆15Aug 20, 2020Updated 5 years ago
- Hackable pure-torch implementation of diffusion models☆23Sep 21, 2024Updated last year
- The code of paper "Toward Optimal LLM Alignments Using Two-Player Games".☆17Jun 20, 2024Updated last year
- FlashSinkhorn: IO-Aware Entropic Optimal Transport in PyTorch + Triton. Streaming Sinkhorn with O(nd) memory.☆31Updated this week
- ☆18Feb 7, 2021Updated 5 years ago
- Proximal Optimal Transport Modeling of Population Dynamics (AISTATS 2022)☆22Jun 19, 2023Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆46Aug 19, 2023Updated 2 years ago
- ☆18Mar 16, 2022Updated 3 years ago
- ☆22Feb 4, 2025Updated last year
- Implementation of the Gromov-Wasserstein distance to the setting of Unbalanced Optimal Transport☆45Mar 13, 2023Updated 2 years ago
- Official PyTorch Implementation for Metric Residual Networks for Sample Efficient Goal-Conditioned Reinforcement Learning☆19Jan 11, 2023Updated 3 years ago
- ☆23Oct 21, 2024Updated last year
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆27Mar 21, 2024Updated last year
- A repo where I play with conditional flow approaches for learning time-varying vector-fields.☆23Jun 4, 2024Updated last year
- OpenAI Gym environment for DART robotics simulator.☆22Apr 17, 2018Updated 7 years ago
- JAX exponential map normalising flows on sphere☆17Oct 4, 2020Updated 5 years ago
- Cross-Domain Imitation Learning via Optimal Transport☆25Jun 24, 2022Updated 3 years ago
- Optimal Transport in the Big Data Era☆116Nov 4, 2024Updated last year
- [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.☆25Apr 15, 2023Updated 2 years ago
- Plannable Approximations to MDP Homomorphisms: Equivariance under Actions☆30Jun 30, 2020Updated 5 years ago
- minimal diffusion model for self-study☆27Jul 8, 2023Updated 2 years ago
- ☆11Sep 18, 2025Updated 4 months ago
- Code for ICLR 2022 paper Rethinking Goal-Conditioned Supervised Learning and Its Connection to Offline RL.☆28Feb 21, 2022Updated 3 years ago
- Code for minimum-entropy coupling.☆32Jan 6, 2026Updated last month
- Repo for the paper "Landscape Surrogate Learning Decision Losses for Mathematical Optimization Under Partial Information"☆38Jul 20, 2023Updated 2 years ago
- LayoutFlow: Flow Matching for Layout Generation [Andrade Guerreiro et al., ECCV 2024]☆35Sep 29, 2025Updated 4 months ago
- TD-Regularized Actor-Critic Methods☆36Dec 26, 2019Updated 6 years ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆41Jun 6, 2024Updated last year
- Understanding Short-Horizon Bias in Stochastic Meta-Optimization☆37Mar 8, 2018Updated 7 years ago
- Getting Started in Imitation Learning☆13Mar 3, 2025Updated 11 months ago
- [NeurIPS 2025] Official code for "Tropical Attention: Neural Algorithmic Reasoning for Combinatorial Algorithms"☆23Oct 23, 2025Updated 3 months ago