moskomule / softdisc

Differentiable Discrete Algorithms for PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

softdisc

Differentiable Discrete Algorithms for PyTorch

🚧🚧The APIs are subject to change🚧🚧

Installation

Requirements

Python>=3.9
PyTorch>=1.8

Installation

pip install -U git+https://github.com/moskomule/softdisc

Algorithms

  • Cuturi, M., Teboul, O., and Vert, J.P., "Differentiable Sorting using Optimal Transport: The Sinkhorn CDF and Quantile Operator." NeurIPS, 2019.
  • Berthet, Q., Blondel, M., Teboul O., Cuturi, M., Vert, J.P., and Bach, F. "Learning with Differentiable Perturbed Optimizers." ICML, 2020.

About

Differentiable Discrete Algorithms for PyTorch

License:MIT License


Languages

Language:Python 100.0%