flox
flox copied to clipboard
Add topk
- [ ] add group starts to
kth - [ ] nantopk & topk; which is public?
- [ ] edit CLAUDE.md
- [ ] add numpy to topk definition
This is pretty ugly:
- Doesn't handle
topkandnantopkyet - Unlike quantile, topk can work as a tree-reduce algorithm. So we have to hack the
kkwarg, passed infinalize_kwargsin many places. This isn't an issue for quantile, because that only works blockwise
Note to myself: this needs some harmonizing of _DUMMY_AXIS is -2 and the fact that the k-axis is 0