GitHub repository: thomgrand/torch_kdtree
Primary language: C++
Description: Port of cp_kdtree to pytorch