https://github.com/graphcore-research/pytorch-approx-topk
Bucketed top-k for PyTorch using a priority queue
https://github.com/graphcore-research/pytorch-approx-topk
kernels pytorch topk
Last synced: about 1 month ago
JSON representation
Bucketed top-k for PyTorch using a priority queue
- Host: GitHub
- URL: https://github.com/graphcore-research/pytorch-approx-topk
- Owner: graphcore-research
- License: mit
- Created: 2024-08-05T17:18:39.000Z (over 1 year ago)
- Default Branch: main
- Last Pushed: 2025-03-22T14:02:35.000Z (11 months ago)
- Last Synced: 2025-03-22T15:19:24.419Z (11 months ago)
- Topics: kernels, pytorch, topk
- Language: Python
- Homepage:
- Size: 9.72 MB
- Stars: 3
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE