jeanfeydy / geomloss

Related projects ⓘ

Alternatives and complementary repositories for geomloss

• gpeyre / SinkhornAutoDiff
  Toolbox to integrate optimal transport loss functions using automatic differentiation and Sinkhorn's algorithm
  ☆431Updated 6 years ago
• lezcano / geotorch
  Constrained optimization toolkit for PyTorch
  ☆657Updated 2 years ago
• dfdazac / wassdistance
  Approximating Wasserstein distances with PyTorch
  ☆455Updated last year
• geoopt / geoopt
  Riemannian Adaptive Optimization Methods with pytorch optim
  ☆850Updated 6 months ago
• getkeops / keops
  KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
  ☆1,054Updated this week
• wgrathwohl / JEM
  Project site for "Your Classifier is Secretly an Energy-Based Model and You Should Treat it Like One"
  ☆417Updated 2 years ago
• jonbarron / robust_loss_pytorch
  A pytorch port of google-research/google-research/robust_loss/
  ☆663Updated 3 years ago
• f-dangel / backpack
  BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
  ☆561Updated 6 months ago
• mfinzi / LieConv
  ☆258Updated last year
• bayesiains / nflows
  Normalizing flows in PyTorch
  ☆845Updated 11 months ago
• mctorch / mctorch
  A manifold optimization library for deep learning
  ☆236Updated 3 years ago
• noahgolmant / pytorch-hessian-eigenthings
  Efficient PyTorch Hessian eigendecomposition tools!
  ☆363Updated 8 months ago
• juho-lee / set_transformer
  Pytorch implementation of set transformer
  ☆545Updated 4 years ago
• rtqichen / ffjord
  code for "FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Models".
  ☆628Updated 4 years ago
• nicola-decao / s-vae-pytorch
  Pytorch implementation of Hyperspherical Variational Auto-Encoders
  ☆351Updated 4 years ago
• jhjacobsen / invertible-resnet
  Official Code for Invertible Residual Networks
  ☆518Updated 3 months ago
• hjmshi / PyTorch-LBFGS
  A PyTorch implementation of L-BFGS.
  ☆586Updated last year
• teddykoker / torchsort
  Fast, differentiable sorting and ranking in PyTorch
  ☆775Updated 10 months ago
• ikostrikov / pytorch-flows
  PyTorch implementations of algorithms for density estimation
  ☆576Updated 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
  ☆605Updated 3 years ago
• josipd / torch-two-sample
  A PyTorch library for two-sample tests
  ☆237Updated last year
• t-vi / pytorch-tvmisc
  Totally Versatile Miscellanea for Pytorch
  ☆468Updated 2 years ago
• locuslab / deq
  [NeurIPS'19] Deep Equilibrium Models
  ☆727Updated 2 years ago
• johannbrehmer / manifold-flow
  Manifold-learning flows (ℳ-flows)
  ☆230Updated 3 years ago
• rusty1s / pytorch_cluster
  PyTorch Extension Library of Optimized Graph Cluster Algorithms
  ☆823Updated 2 months ago
• rtqichen / residual-flows
  code for "Residual Flows for Invertible Generative Modeling".
  ☆261Updated last year
• didriknielsen / survae_flows
  Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"
  ☆283Updated 3 years ago
• y0ast / Glow-PyTorch
  Simple, extendable, easy to understand Glow implementation in PyTorch
  ☆376Updated 2 years ago
• aliutkus / torchsearchsorted
  Pytorch Custom CUDA kernel for searchsorted
  ☆135Updated last year