Fast kd-tree implementation in Python