pytorch
GPU tensor library
A Python library providing tensors and dynamic neural networks with strong GPU acceleration
Tensors and Dynamic neural networks in Python with strong GPU acceleration
85k stars
2k watching
23k forks
Language: Python
last commit: 11 months ago
Linked from 10 awesome lists
autograddeep-learninggpumachine-learningneural-networknumpypythontensor
Backlinks from these awesome lists:
-
vinta/awesome-python
-
mungell/awesome-for-beginners
-
josephmisiti/awesome-machine-learning
-
fffaraz/awesome-cpp
-
academic/awesome-datascience
-
christoschristofidis/awesome-deep-learning
-
ethicalml/awesome-production-machine-learning
-
jobbole/awesome-cpp-cn
-
krzjoa/awesome-python-data-science
-
rootsongjc/awesome-cloud-native
Related projects:
| Repository | Description | Stars |
|---|---|---|
 yunjey/pytorch-tutorial | A comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage. | 30,401 |
 victoresque/pytorch-template | An open-source PyTorch deep learning project template providing a structured environment for building and training various models. | 4,784 |
| pytorch/examples | A collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning. | 22,530 |
 pyg-team/pytorch_geometric | A PyTorch-based library for training and applying Graph Neural Networks to structured data | 21,597 |
 kakaobrain/torchgpipe | A PyTorch-based library for efficient training of large neural networks using pipeline parallelism and automatic recomputation of gradients. | 820 |
 qubvel-org/segmentation_models.pytorch | A comprehensive library for training and applying deep learning models for image segmentation | 9,829 |
 pkmital/tensorflow_tutorials | A collection of examples and tutorials illustrating various applications and use cases of TensorFlow in Python | 5,636 |
 szagoruyko/pytorchviz | A tool to visualize PyTorch execution graphs and traces | 3,251 |
 nvidia/flownet2-pytorch | A deep learning framework implementing an optical flow estimation model using PyTorch. | 3,159 |
 wkentaro/pytorch-for-numpy-users | Provides a PyTorch interface for working with NumPy arrays and tensors. | 691 |
 sniklaus/pytorch-extension | A PyTorch CUDA extension that uses CuPy to compute the Hadamard product of two tensors. | 117 |
 tobiascz/mnist_pytorch_python_and_capi | An example project demonstrating how to train and deploy a neural network in Python and C++ using PyTorch 1.0 | 96 |
| nvidia/minkowskiengine | An auto-differentiation library for sparse tensors used in computer vision and deep learning applications. | 2,513 |
 jwyang/faster-rcnn.pytorch | An object detection implementation built on top of PyTorch, supporting multi-image batch training and multiple GPUs. | 7,721 |
| pytorch/ignite | A high-level library to help with training and evaluating neural networks in PyTorch | 4,554 |