NVIDIA/MinkowskiEngine
Auto-differentiation library for high-dimensional sparse tensors used to build neural networks with sparse convolution operations.

The Minkowski Engine is a PyTorch-based neural network library specializing in sparse tensor operations and auto-differentiation for high-dimensional data. It provides standard neural network layers such as convolution, pooling, unpooling, and broadcasting for sparse tensors, with CUDA acceleration. The library targets applications in 3D computer vision, semantic segmentation, and spatio-temporal analysis, enabling efficient deep learning on point clouds and volumetric data.