[Feature Request] High Efficient Shortest Path Implementation

microsoft / Graphormer

Graphormer is a general-purpose deep learning backbone for molecular modeling.

MIT License

2.08k stars 334 forks source link

[Feature Request] High Efficient Shortest Path Implementation #69

Open zhengsx opened 2 years ago

zhengsx commented 2 years ago

Summary

Calculation of shortest path on CPU would be the bottlenect when batchsize is small. High efficient implementation comparing to current Cython is desired.