accelerating accelerators with differentiable kD-trees - View it on GitHub