/successive-halving-topk

A fast and highly accurate differentiable Top-k operator from the "Successive Halving Top-k Operator" AAAI'21 paper.

Primary LanguagePythonMIT LicenseMIT

Stargazers