Node2Vec with raw PyTorch

[bokveizen]

Hi, currently I am using a non-CUDA device. Specifically, a device with Intel Gaudi HPUs.
When I tried to use Node2Vec, I realized that it requires either pyg_lib or torch_cluster, but both libraries only support CPU or CUDA GPUs.
Essentially, we need the random_walk function in torch_cluster.
I am wondering whether it is possible to have an implementation of random_walk in raw Pytorch?
If not, what should I do to make it possible?

Thank you for your help in advance.