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
8.8K
2 Sources, first seen ago