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

Awesome Lists containing this project