ot-triton-lab / flash-sinkhornView external linksLinks
FlashSinkhorn: IO-Aware Entropic Optimal Transport in PyTorch + Triton. Streaming Sinkhorn with O(nd) memory.
☆31Updated this week
Alternatives and similar repositories for flash-sinkhorn
Users that are interested in flash-sinkhorn are comparing it to the libraries listed below
Sorting:
- Proximal Optimal Transport Modeling of Population Dynamics (AISTATS 2022)☆22Jun 19, 2023Updated 2 years ago
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆37Mar 18, 2025Updated 11 months ago
- The grid dataset, provided by DFG research unit FOR2083☆15Aug 3, 2023Updated 2 years ago
- Causal Impact of an intervention integrated with control group selection☆10Sep 11, 2022Updated 3 years ago
- A toolkit to bulk-submission, download, analysis and summarizing data from abYsis.org☆10Apr 20, 2020Updated 5 years ago
- Simulation and inference of gene regulatory networks based on transcriptional bursting☆12Oct 26, 2025Updated 3 months ago
- This is the class in matlab for convex optimization algorithms☆10Nov 19, 2023Updated 2 years ago
- 在R语言中使用ChatGPT（中国大陆版）☆12Mar 17, 2023Updated 2 years ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆42May 8, 2022Updated 3 years ago
- A Course on designing high-throughput experiments☆12Sep 25, 2024Updated last year
- Repository for PlasticParcels tool☆16Updated this week
- This repository implements a Vision Transformer (ViT) to classify handwritten digits from the MNIST dataset. The project includes model d…☆15May 29, 2024Updated last year
- A custom git utility for managing file sizes in your git workflow.☆11Jan 17, 2022Updated 4 years ago
- Principal Geodesic Analysis in the Wasserstein space☆10Jun 19, 2018Updated 7 years ago
- A python implementation of discrete optimal transport with a Tsallis entropy regularization.☆14Oct 23, 2023Updated 2 years ago
- Advanced Quantitative Factor Research: ML-powered stock return prediction with 72% performance improvement. Features comprehensive alpha …☆385Aug 22, 2025Updated 5 months ago
- A Python package for street view image perception analysis, providing tools for feature extraction and comfort prediction.☆82Apr 17, 2025Updated 10 months ago
- ☆24Feb 6, 2026Updated last week
- Graphical modelling with time series data using an ODE model☆12Mar 28, 2022Updated 3 years ago
- ⛔️ DEPRECATED GPU Ocean Python/CUDA codebase☆11Nov 9, 2023Updated 2 years ago
- A simple implication of the paper "FedCTQ: A Federated-based Framework for Accurate and Efficient Contact Tracing Query"☆70Nov 9, 2025Updated 3 months ago
- Human embryonic cell altas☆13Dec 31, 2019Updated 6 years ago
- Code repository for paper: A score-based filter for nonlinear data assimilation☆13Nov 5, 2024Updated last year
- Artificial Intelligence in Transportation☆17Jun 6, 2020Updated 5 years ago
- 提供一系列组件提供快速高效大家高质量中后台应用的能力☆18Aug 21, 2025Updated 5 months ago
- Reading Group @mila-iqia on Computational Optimal Transport for Machine Learning Applications☆13Jun 3, 2022Updated 3 years ago
- PYthon RAndom SAmpling for MEshes☆15Feb 2, 2026Updated 2 weeks ago
- Project bulk/single-cell RNA-seq data to given LSI space☆11Jan 17, 2022Updated 4 years ago
- GPU accelerated particle filter☆13Apr 28, 2021Updated 4 years ago
- Python API and analysis of Chicago's bikeshare☆10Dec 8, 2022Updated 3 years ago
- PyTorch implementation of "Robust Barycenter Estimation using Semi-unbalanced Neural Optimal Transport" (ICLR 2025)☆17Jul 7, 2025Updated 7 months ago
- Predict traffic collisions in Calgary using Keras and Tensorflow.js. Visualization with React and Mapbox.☆12Feb 21, 2019Updated 6 years ago
- 本项目是一款结合15 个主流平台的 26 个榜单实时数据与大模型分析能力的舆情分析助手。通过前端页面，用户可实现对话式热搜榜单查询、特定主题搜索、话题聚类分析及情感倾向分析。系统支持快捷键控制爬虫启停、多平台数据快速查询与跳转，并能基于新闻详情页内容（即使是视频信息也能挖掘…☆331Jan 4, 2026Updated last month
- A small r package to provide the basic last.year analytics also performed by last.fm☆12Dec 8, 2019Updated 6 years ago
- ☆19Mar 31, 2023Updated 2 years ago
- Polynomials for efficiently computing the density of seawater☆15Jan 12, 2026Updated last month
- NATMI: Network Analysis Toolkit for the Multicellular Interactions☆17Dec 19, 2024Updated last year
- ☆13Jul 28, 2023Updated 2 years ago
- Pytorch implementation of "Light Schrödinger Bridge" (ICLR 2024)☆64Aug 9, 2024Updated last year