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:
- This repository contains a JAX implementation of Πnet, an output layer for neural networks that ensures the satisfaction of specified con…☆67Aug 19, 2026Updated 2 months ago
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆33Mar 21, 2024Updated 2 years ago
- Repository detailing updated allowlisting and clonal analysis workflow☆11Feb 1, 2025Updated last year
- ☆49Aug 19, 2026Updated 2 months ago
- Tools for image processing, brain segmentation and evaluation of glioblastoma growth models☆14Updated this week
- 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.
- Official implementation of Graph Neural Networks as Gradient Flows.☆15Aug 15, 2023Updated 3 years ago
- E(3)-Equivariant Mesh Neural Networks (AISTATS 2024)☆12Apr 24, 2024Updated 2 years ago
- Fair and objective benchmarking of diffusion posterior sampling algorithms for Bayesian inverse problems.☆13Sep 17, 2025Updated last year
- PyTorch implementation of "Field Matching: an Electrostatic Paradigm to Generate and Transfer Data" (ICML 2025)☆21Jul 23, 2025Updated last year
- Optimal Control Tutorials☆14Aug 20, 2020Updated 6 years ago
- Local Agentic Model Deployment on AI Cluster☆19Updated this week
- DeepRUOTv2 (Dynamical Optimal Transport Tools)☆23Dec 10, 2025Updated 10 months ago
- Cluster-Normalize-Activate Modules☆13Jan 13, 2025Updated last year
- ChebLieNet, a spectral graph neural network turned equivariant by Riemannian geometry on Lie groups.☆14Aug 20, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Optimal Transport in the Big Data Era☆118Nov 4, 2024Updated 2 years ago
- Implementations of 'Discrete-state Continuous-time Diffusion for Graph Generation', NeurIPS'24.☆14Oct 5, 2025Updated last year
- Some standard pieces I find useful in Typst: fonts, subfigures, algorithms, labeled enums, etc.☆13Oct 16, 2026Updated 3 weeks ago
- On Information-Theoretic Measures of Predictive Uncertainty☆12Oct 15, 2024Updated 2 years ago
- ☆16Nov 21, 2025Updated 11 months ago
- Maros-Meszaros test set to benchmark QP solvers☆10Oct 1, 2026Updated last month
- Code for the Poincaré Wasserstein Autoencode paper☆10Sep 25, 2019Updated 7 years ago
- ☆23Feb 4, 2025Updated last year
- PyTorch implementation of "Adversarial Schrödinger Bridge Matching" (NeurIPS 2024)☆31Oct 15, 2026Updated 3 weeks 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.
- ☆13Jan 20, 2023Updated 3 years ago
- Official code for the MICCAI 2025 paper "Semantically Consistent Discrete Diffusion for 3D Biological Graph Generation"☆19Jul 7, 2025Updated last year
- Implementation of deep kernels in GPyTorch and Pyro☆13Jan 19, 2021Updated 5 years ago
- cp2k-2-deepmdkit (c2d): A comprehensive tool for converting cp2k output to DeepMD-kit input☆10Aug 19, 2024Updated 2 years ago
- Official code repository to the corresponding paper.☆29Sep 14, 2023Updated 3 years ago
- ☆16Aug 24, 2026Updated 2 months ago
- Paper: Fast and accurate prediction of partial charges using Atom-Path-Descriptor-based machine learning☆10Jun 26, 2020Updated 6 years ago
- Temperature map for Leaflet inspired by https://github.com/ham-systems/temperature-map-gl☆10Jan 6, 2023Updated 3 years ago
- Fast and reliable distributed systems in Python☆36Oct 25, 2026Updated last week
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Python API and analysis of Chicago's bikeshare☆10Dec 8, 2022Updated 3 years ago
- [NeurIPS 2026] Official implementation of Contrastive Distribution Matching for Amortized Sequential Monte Carlo in Discrete Diffusion☆15Oct 1, 2026Updated last month
- ☆11Jun 1, 2025Updated last year
- Automate Finding and Fetching Academic Citations☆11Oct 10, 2024Updated 2 years ago
- SWOT-Flow official paper implementation☆10Jul 26, 2022Updated 4 years ago
- Official Pytorch implementation of "Neural Optimal Transport with General Cost Functionals" (ICLR 2024)☆24Aug 29, 2024Updated 2 years ago
- ☆26Oct 10, 2026Updated 3 weeks ago