JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".
☆40Mar 18, 2025Updated last year
Alternatives and similar repositories for jkonet-star
Users that are interested in jkonet-star are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆41May 8, 2022Updated 4 years ago
- Fast Laplacian estimation☆17May 31, 2024Updated 2 years ago
- Implementation of the Gromov-Wasserstein distance to the setting of Unbalanced Optimal Transport☆50Mar 13, 2023Updated 3 years ago
- Repository detailing updated allowlisting and clonal analysis workflow☆11Feb 1, 2025Updated last year
- TopoTrans: Optimal Transport meets Topological Data Analysis☆14Apr 20, 2023Updated 3 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.
- Python numerical optimization toolbox☆12Nov 20, 2018Updated 7 years ago
- Cluster-Normalize-Activate Modules☆13Jan 13, 2025Updated last year
- It is a repo which allows to compute all divergences derived from the theory of entropically regularized, unbalanced optimal transport. I…☆32Dec 21, 2022Updated 3 years ago
- Code and data for paper named: Large language models for automatic equation discovery of nonlinear dynamics☆14Mar 6, 2025Updated last year
- Tutorial (in Japanese) on linear cross entropy benchmark using qulacs.☆12Dec 16, 2019Updated 6 years ago
- DeepRUOTv2 (Dynamical Optimal Transport Tools)☆24Dec 10, 2025Updated 9 months ago
- Optimal Transport in the Big Data Era☆118Nov 4, 2024Updated last year
- Euclidean Wasserstein-2 optimal transportation☆46Aug 19, 2023Updated 3 years ago
- PyTorch implementation of "Field Matching: an Electrostatic Paradigm to Generate and Transfer Data" (ICML 2025)☆21Jul 23, 2025Updated last year
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- ☆23Feb 4, 2025Updated last year
- ☆17Mar 16, 2022Updated 4 years ago
- Optimal Control Tutorials☆15Aug 20, 2020Updated 6 years ago
- JAX exponential map normalising flows on sphere☆17Oct 4, 2020Updated 5 years ago
- Code release for "Stochastic Optimal Control Matching"☆43Aug 14, 2024Updated 2 years ago
- ChebLieNet, a spectral graph neural network turned equivariant by Riemannian geometry on Lie groups.☆14Aug 20, 2024Updated 2 years ago
- An Efficient Convex Optimization-based Framework for Signal Temporal Logic (STL) Specifications☆17Jun 7, 2023Updated 3 years ago
- PyTorch implementation of "Adversarial Schrödinger Bridge Matching" (NeurIPS 2024)☆30Jun 2, 2025Updated last year
- Input an origin point, generate a heatmap of driving times to surrounding areas at different times of day. 🚗 ⏱ 🔥 🗺☆10Apr 17, 2022Updated 4 years 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.
- Source files for our regularization paper!☆15May 11, 2019Updated 7 years ago
- Causal Impact of an intervention integrated with control group selection☆10Sep 11, 2022Updated 4 years ago
- Optimal Transport for Machine Learners☆89Aug 26, 2026Updated last month
- Chance constraints in CVXPY☆19Jan 12, 2019Updated 7 years ago
- This is official repository for "ConStyle v2: A Strong Prompter for All-in-One Image Restoration"☆13Jun 12, 2024Updated 2 years ago
- Python API and analysis of Chicago's bikeshare☆10Dec 8, 2022Updated 3 years ago
- Official Implementation of "Graph Neural Network-Inspired Kernels for Gaussian Processes in Semi-Supervised Learning"☆18Feb 11, 2023Updated 3 years ago
- Program Package for Sampling, Training and Applying ML-based Potential models☆13Aug 13, 2026Updated last month
- The codes of PSF-aware Aberration-image Restoration Transformer (PART).☆14Sep 25, 2025Updated last year
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Official code repository to the corresponding paper.☆29Sep 14, 2023Updated 3 years ago
- Convex hulls of reachable sets☆19Apr 14, 2026Updated 5 months ago
- Official Pytorch implementation of "Neural Optimal Transport with General Cost Functionals" (ICLR 2024)☆24Aug 29, 2024Updated 2 years ago
- Implicit Differentiable Optimal Control (IDOC) with JAX☆12May 11, 2022Updated 4 years ago
- Cross-Domain Imitation Learning via Optimal Transport☆27Jun 24, 2022Updated 4 years ago
- Mutual information estimators and benchmarks☆14Aug 13, 2026Updated last month
- Model Predictive Control for Signal Temporal Logic with Time Interval Decomposition☆14May 26, 2024Updated 2 years ago