Search Packages

point-cloud-utils

An easy-to-use Python library for processing and manipulating 3D point clouds and meshes.

ott-jax

Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.

mhc-mlx

MLX + Metal implementation of mHC: Manifold-Constrained Hyper-Connections by DeepSeek-AI.

torchgw

Fast Sampled Gromov-Wasserstein optimal transport solver — pure PyTorch, scalable, differentiable

flash-sinkhorn

Sinkhorn optimal transport kernels in PyTorch + Triton (squared Euclidean, no cost matrix materialization).