JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".
☆37Mar 18, 2025Updated 11 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:
- Implicit Differentiable Optimal Control (IDOC) with JAX☆12May 11, 2022Updated 3 years ago
- TopoTrans: Optimal Transport meets Topological Data Analysis☆14Apr 20, 2023Updated 2 years ago
- Fast Laplacian estimation☆16May 31, 2024Updated last year
- ☆20Dec 7, 2025Updated 3 months ago
- [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.☆26Jul 4, 2025Updated 8 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.☆63Dec 19, 2025Updated 2 months ago
- 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
- 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
- ☆24Oct 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
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆31Mar 21, 2024Updated last year
- Optimal Transport in the Big Data Era☆117Nov 4, 2024Updated last year
- [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.☆25Apr 15, 2023Updated 2 years ago
- minimal diffusion model for self-study☆28Jul 8, 2023Updated 2 years ago
- Plannable Approximations to MDP Homomorphisms: Equivariance under Actions☆30Jun 30, 2020Updated 5 years ago
- Tools for visualizing and comparing data from vertebrate retinas☆14Jan 20, 2025Updated last year
- Code for minimum-entropy coupling.☆32Jan 6, 2026Updated 2 months ago
- Cluster-Normalize-Activate Modules☆13Jan 13, 2025Updated last year
- LayoutFlow: Flow Matching for Layout Generation [Andrade Guerreiro et al., ECCV 2024]☆35Sep 29, 2025Updated 5 months ago
- Repo for the paper "Landscape Surrogate Learning Decision Losses for Mathematical Optimization Under Partial Information"☆38Jul 20, 2023Updated 2 years ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆41Jun 6, 2024Updated last year
- Corax: Core RL in JAX☆40Feb 22, 2024Updated 2 years ago
- My CV☆39Updated this week
- Understanding Short-Horizon Bias in Stochastic Meta-Optimization☆37Mar 8, 2018Updated 8 years ago
- Tree-structured recurrent switching linear dynamical systems☆38Jul 13, 2020Updated 5 years ago
- [NeurIPS 2025] Official code for "Tropical Attention: Neural Algorithmic Reasoning for Combinatorial Algorithms"☆23Oct 23, 2025Updated 4 months ago
- This is the class in matlab for convex optimization algorithms☆10Nov 19, 2023Updated 2 years ago
- The grid dataset, provided by DFG research unit FOR2083☆15Aug 3, 2023Updated 2 years ago
- Package to design primers for MutaSeq and related methods☆11Jan 15, 2021Updated 5 years ago
- ☆10Jun 25, 2021Updated 4 years ago