GitHub repository: thomgrand/tf_kdtree
Primary language: C++
Description: This repository implements the KDTree on CUDA with interface bindings both in C++ and Tensorflow