PythonOT / POT

Related projects ⓘ

Alternatives and complementary repositories for POT

• jeanfeydy / geomloss
  Geometric loss functions between point clouds, images and volumes
  ☆596Updated 10 months ago
• lezcano / geotorch
  Constrained optimization toolkit for PyTorch
  ☆661Updated 2 years ago
• gpeyre / SinkhornAutoDiff
  Toolbox to integrate optimal transport loss functions using automatic differentiation and Sinkhorn's algorithm
  ☆434Updated 6 years ago
• geoopt / geoopt
  Riemannian Adaptive Optimization Methods with pytorch optim
  ☆853Updated 6 months ago
• cvxgrp / cvxpylayers
  Differentiable convex optimization layers
  ☆1,823Updated last week
• getkeops / keops
  KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
  ☆1,056Updated this week
• cornellius-gp / gpytorch
  A highly efficient implementation of Gaussian Processes in PyTorch
  ☆3,582Updated this week
• ldeecke / gmm-torch
  Gaussian mixture models in PyTorch.
  ☆518Updated last year
• pymanopt / pymanopt
  Python toolbox for optimization on Riemannian manifolds with support for automatic differentiation
  ☆766Updated last week
• geomstats / geomstats
  Computations and statistics on manifolds with geometric structures.
  ☆1,269Updated last week
• ott-jax / ott
  Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
  ☆527Updated this week
• bayesiains / nflows
  Normalizing flows in PyTorch
  ☆848Updated last year
• szagoruyko / pytorchviz
  A small package to create visualizations of PyTorch execution graphs
  ☆3,236Updated 7 months ago
• jettify / pytorch-optimizer
  torch-optimizer -- collection of optimizers for Pytorch
  ☆3,044Updated 8 months ago
• optimaltransport / optimaltransport.github.io
  Web site of the Computational Optimal Transport book
  ☆356Updated last year
• janosh / awesome-normalizing-flows
  Awesome resources on normalizing flows.
  ☆1,438Updated last month
• dfdazac / wassdistance
  Approximating Wasserstein distances with PyTorch
  ☆455Updated last year
• DiffEqML / torchdyn
  A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods
  ☆1,400Updated 6 months ago
• clementchadebec / benchmark_VAE
  Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)
  ☆1,824Updated 3 months ago
• facebookresearch / higher
  higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…
  ☆1,594Updated 2 years ago
• google-research / disentanglement_lib
  disentanglement_lib is an open-source library for research on learning disentangled representations.
  ☆1,389Updated 3 years ago
• teddykoker / torchsort
  Fast, differentiable sorting and ranking in PyTorch
  ☆774Updated 10 months ago
• locuslab / qpth
  A fast and differentiable QP solver for PyTorch.
  ☆687Updated 2 months ago
• kilianFatras / awesome-optimal-transport
  A list of awesome papers and cool resources on optimal transport and its applications in general! As you will notice, this list is curren…
  ☆206Updated 3 years ago
• vislearn / FrEIA
  Framework for Easily Invertible Architectures
  ☆773Updated 4 months ago
• skorch-dev / skorch
  A scikit-learn compatible neural network library that wraps PyTorch
  ☆5,885Updated 2 weeks ago
• CannyLab / tsne-cuda
  GPU Accelerated t-SNE for CUDA with Python bindings
  ☆1,804Updated last month
• lmcinnes / umap
  Uniform Manifold Approximation and Projection
  ☆7,468Updated 2 weeks ago
• google / neural-tangents
  Fast and Easy Infinite Neural Networks in Python
  ☆2,279Updated 8 months ago
• cvxgrp / pymde
  Minimum-distortion embedding with PyTorch
  ☆538Updated last year