#177 · Primary category: Deep Learning Frameworks
MinkowskiEngine
Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
Project last updated:03/05/24
GitHub Stars
3.0K
Forks
484
Contributors
25
License
Other
Why we included this project
Most 3D perception work starts with data that is mostly empty space, and MinkowskiEngine is built around that fact instead of fighting it. It is an auto-differentiation library for sparse tensors, so convolution, pooling, unpooling, and broadcasting all operate directly on the points that actually exist, skipping the empty voxels a dense tensor would force you to pad. That keeps memory and training time down for LiDAR scans and large 3D scenes. It slots into PyTorch, ships CUDA-accelerated kernels, and includes runnable examples for indoor segmentation, classification, and reconstruction, which is handy for checking how it behaves before you commit a project to it. The main friction is installation: expect a CUDA build step and some care matching PyTorch versions.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.