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
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