MinkowskiEngine

Tensor processing library

An auto-differentiation library for sparse tensors used in computer vision and deep learning applications.

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

GitHub

3k stars
46 watching
369 forks
Language: Python
last commit: 10 months ago
3d-convolutional-network3d-vision4d-convolutional-neural-networkauto-differentiationcomputer-visionconvolutional-neural-networkscudadeep-learninghigh-dimensional-datahigh-dimensional-inferenceminkowski-engineneural-networkpytorchsemantic-segmentationspace-timesparse-convolutionsparse-tensor-networksparse-tensorsspatio-temporal-analysistrilateral-filter

Related projects:

Repository Description Stars
pytorch/pytorch A Python library providing tensors and dynamic neural networks with strong GPU acceleration 84,978
pkmital/tensorflow_tutorials A collection of examples and tutorials illustrating various applications and use cases of TensorFlow in Python 5,636
nvidia/flownet2-pytorch A deep learning framework implementing an optical flow estimation model using PyTorch. 3,159
nvidia/tensorrt Provides a set of tools and libraries for optimizing deep learning inference on NVIDIA GPUs. 10,926
tensorpack/tensorpack A high-performance neural network training interface for TensorFlow that focuses on speed and flexibility. 6,303
sjchoi86/tensorflow-101 A comprehensive tutorial project on building and training neural networks with TensorFlow 2,606
nvidia/fastertransformer A high-performance transformer-based NLP component optimized for GPU acceleration and integration into various frameworks. 5,937
leriomaggio/deep-learning-keras-tensorflow A comprehensive tutorial on building and training deep neural networks using Keras and TensorFlow 2,951
lanpa/tensorboardx A tool for visualizing and logging training progress during deep learning model development. 7,887
tensorflow/lingvo A software framework for building sequence models using neural networks in TensorFlow 2,820
binroot/tensorflow-book A comprehensive resource for learning machine learning using TensorFlow. 4,453
sony/nnabla A deep learning framework that provides a flexible and expressive Python API for building and training neural networks on various platforms. 2,729
facebookresearch/pytorch3d A deep learning library for 3D data processing and computer vision research using PyTorch 8,889
arogozhnikov/einops A flexible and powerful library for performing tensor operations in various deep learning frameworks. 8,574
pyg-team/pytorch_geometric A PyTorch-based library for training and applying Graph Neural Networks to structured data 21,597