https://github.com/srush/torch-queue

https://github.com/srush/torch-queue

Last synced: 11 months ago
JSON representation

• Host: GitHub
• URL: https://github.com/srush/torch-queue
• Owner: srush
• License: mit
• Created: 2022-02-08T15:33:03.000Z (over 4 years ago)
• Default Branch: main
• Last Pushed: 2023-11-09T13:15:09.000Z (over 2 years ago)
• Last Synced: 2025-04-12T15:13:30.764Z (about 1 year ago)
• Language: Python
• Size: 20.5 KB
• Stars: 67
• Watchers: 4
• Forks: 4
• Open Issues: 2
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# torch-queue v0.1

A protoype batched differentiable priority queue in raw pytorch, and a simple djikstra implementation.  

Roughly based on [BGPQ: A Heap-Based Priority Queue Design for GPUs](https://dl.acm.org/doi/fullHtml/10.1145/3472456.3472463)

![image](https://user-images.githubusercontent.com/35882/153462163-c0013536-879c-4779-82b1-1fac4704e987.png)