Python Library for Signal/Image Data Analysis with Transport Methods

arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv

ICLR22 "Fast Differentiable Matrix Square Root" and T-PAMI extension

Matrix square root with gradient support for PyTorch

BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.

Course to get into Large Language Models (LLMs) with roadmaps and Colab notebooks.

Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

A reactive notebook for Python — run reproducible experiments, execute as a script, deploy as an app, and version with git.

Official Implementation for The Numerical Stability of Hyperbolic Representation Learning

Code for Supervised Word Mover's Distance (SWMD)

TorchCFM: a Conditional Flow Matching library

Hyperbolic Embeddings

This is the official GitHub repo for "Deep Optimal Transport for Domain Adaptation on SPD Manifolds"

Computations and statistics on manifolds with geometric structures.

About A collection of AWESOME things about information geometry Topics

Differentiable spherical harmonic transforms and spherical convolutions in PyTorch

A package for Multiple Kernel Learning in Python

A Python library for variational inference with normalizing flow and annealing

An extension of the PyTorch library containing various tools for performing deep learning in hyperbolic space.

Embedding graphs in symmetric spaces

Distributional Sliced-Wasserstein distance code

Fast Laplacian estimation

Normalizing-flow enhanced sampling package for probabilistic inference in Jax

code for CO-SNE: Dimensionality Reduction and Visualization for Hyperbolic Data (CVPR 2022)