### 📚 Describe the documentation issue Hi, I saw the source in the doc using KDTree for CPU and assumed it'd be the same on the GPU. But the CUDA code is brute-force. I hope this is clarified in the doc. ### Suggest a potential alternative/fix _No response_