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

2k stars
46 watching
368 forks
Language: Python
last commit: 9 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 83,959
pkmital/tensorflow_tutorials A collection of examples and tutorials illustrating various applications and use cases of TensorFlow in Python 5,642
nvidia/flownet2-pytorch A deep learning framework implementing an optical flow estimation model using PyTorch. 3,147
nvidia/tensorrt A high-performance deep learning inference platform on NVIDIA GPUs 10,807
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,886
leriomaggio/deep-learning-keras-tensorflow A comprehensive tutorial on building and training deep neural networks using Keras and TensorFlow 2,948
lanpa/tensorboardx A tool for visualizing and logging the progress of machine learning models during training 7,870
tensorflow/lingvo A software framework for building sequence models using neural networks in TensorFlow 2,816
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,728
facebookresearch/pytorch3d A deep learning library for 3D data processing and computer vision research using PyTorch 8,806
arogozhnikov/einops A flexible and powerful library for performing tensor operations in various deep learning frameworks. 8,518
pyg-team/pytorch_geometric A PyTorch-based library for training and applying Graph Neural Networks to structured data 21,382