jeanfeydy / geomloss

Related projects ⓘ

Alternatives and complementary repositories for geomloss

• 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
• getkeops / keops
  KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
  ☆1,054Updated this week
• dfdazac / wassdistance
  Approximating Wasserstein distances with PyTorch
  ☆455Updated last year
• hjmshi / PyTorch-LBFGS
  A PyTorch implementation of L-BFGS.
  ☆586Updated last year
• bayesiains / nflows
  Normalizing flows in PyTorch
  ☆848Updated last year
• geoopt / geoopt
  Riemannian Adaptive Optimization Methods with pytorch optim
  ☆851Updated 6 months ago
• mctorch / mctorch
  A manifold optimization library for deep learning
  ☆236Updated 3 years ago
• fwilliams / scalable-pytorch-sinkhorn
  ☆137Updated 2 years ago
• kamenbliznashki / normalizing_flows
  Pytorch implementations of density estimation algorithms: BNAF, Glow, MAF, RealNVP, planar flows
  ☆606Updated 3 years ago
• didriknielsen / survae_flows
  Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"
  ☆284Updated 3 years ago
• wgrathwohl / JEM
  Project site for "Your Classifier is Secretly an Energy-Based Model and You Should Treat it Like One"
  ☆417Updated 2 years ago
• jhjacobsen / invertible-resnet
  Official Code for Invertible Residual Networks
  ☆519Updated 3 months ago
• ikostrikov / pytorch-flows
  PyTorch implementations of algorithms for density estimation
  ☆577Updated 3 years ago
• rtqichen / ffjord
  code for "FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Models".
  ☆629Updated 4 years ago
• f-dangel / backpack
  BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
  ☆561Updated this week
• juho-lee / set_transformer
  Pytorch implementation of set transformer
  ☆545Updated 4 years ago
• noahgolmant / pytorch-hessian-eigenthings
  Efficient PyTorch Hessian eigendecomposition tools!
  ☆364Updated 8 months ago
• lucidrains / siren-pytorch
  Pytorch implementation of SIREN - Implicit Neural Representations with Periodic Activation Function
  ☆469Updated last year
• johannbrehmer / manifold-flow
  Manifold-learning flows (ℳ-flows)
  ☆230Updated 4 years ago
• rtqichen / residual-flows
  code for "Residual Flows for Invertible Generative Modeling".
  ☆263Updated last year
• tonyduan / normalizing-flows
  Neural Spline Flow, RealNVP, Autoregressive Flow, 1x1Conv in PyTorch.
  ☆272Updated 11 months ago
• mfinzi / LieConv
  ☆258Updated last year
• jonbarron / robust_loss_pytorch
  A pytorch port of google-research/google-research/robust_loss/
  ☆664Updated 3 years ago
• bayesiains / nsf
  Code for Neural Spline Flows paper
  ☆259Updated 4 years ago
• VincentStimper / normalizing-flows
  PyTorch implementation of normalizing flow models
  ☆723Updated 2 months ago
• ott-jax / ott
  Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
  ☆526Updated this week
• KinglittleQ / torch-batch-svd
  A 100x faster SVD for PyTorch⚡️
  ☆447Updated 2 years ago
• teddykoker / torchsort
  Fast, differentiable sorting and ranking in PyTorch
  ☆774Updated 10 months ago
• amir-abdi / disentanglement-pytorch
  Disentanglement library for PyTorch
  ☆274Updated 2 years ago