Hi there! Thanks for this great repo.

I have a feature request, inspired by the k-d tree in scipy:
Could there be a function that finds all points in a KdTree other within a dist for all points in a KdTree self? So basically the scipy function query_ball_tree.

Thanks.