Report
A claimed faster, bitwise-correct fix for Pallas TPU's top-k implementation
The author says the implementation retains a known bug because a fix was deemed expensive.
TLDR
The author says Pallas TPU's top-k implementation retains a known bug because a fix was deemed expensive. They claim to have made it bitwise correct and faster, and link to a Zenodo DOI.
Combined views
178
2 Sources, first seen ago
3 likes2 saves2 reposts