thomgrand/torch_kdtree

Fast kd-tree implementation in Python

Minimal JAX implementation of k-nearest neighbors using a k-d tree.

Benchmarks of approximate nearest neighbor libraries in Python

A Python nearest neighbor descent for approximate nearest neighbors

the official implementation of PointCNN++ (CVPR 2026)

PyTorch Point Cloud models, scripts, pretrained weights -- PointNet++, KPConv, RandLA-Net, VoteNet, SPVCNN, 3DETR, PointTransformer, OctFormer, PointMamba and more

Marching cubes implementation for PyTorch environment.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Make your wildest 3D ConvNet dream architectures come true