pyinn

CuPy ops

A Python implementation of PyTorch neural networks operations using CuPy.

CuPy fused PyTorch neural networks ops

GitHub

274 stars
12 watching
38 forks
Language: Python
last commit: almost 7 years ago
cupypytorch

Related projects:

Repository Description Stars
pochih/fcn-pytorch A Python implementation of fully convolutional networks for semantic segmentation in computer vision. 409
szagoruyko/binary-wide-resnet An implementation of a 1-bit weight neural network architecture using PyTorch 124
yunlongdong/fcn-pytorch A PyTorch implementation of FCN for semantic segmentation with an easy-to-use interface and pre-trained models. 161
sniklaus/pytorch-extension A PyTorch CUDA extension that uses CuPy to compute the Hadamard product of two tensors. 117
koz4k/dni-pytorch Decoupled Neural Interfaces using Synthetic Gradients for PyTorch 236
huguyuehuhu/hcn-pytorch Replication of a PyTorch model for action recognition and detection from skeleton data 218
ikostrikov/pytorch-trpo A PyTorch implementation of an optimization algorithm for continuous control and reinforcement learning tasks 435
gunhochoi/kind-pytorch-tutorial A comprehensive tutorial on building and training PyTorch models using Python 392
kazuto1011/pspnet-pytorch Re-implementation of a deep learning model for semantic segmentation using PyTorch. 52
kaiyangzhou/dassl.pytorch A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. 1,236
fgxaos/pytorch-innvestigate PyTorch implementation of an explainability technique for deep neural networks 9
zsef123/efficientnets-pytorch A PyTorch implementation of EfficientNet for computer vision tasks 309
chingyaoc/ggnn.pytorch An implementation of a neural network architecture for processing graph-structured data and making predictions on nodes. 466
bamos/densenet.pytorch An implementation of a deep neural network architecture in PyTorch 833
zhanghang1989/pytorch-encoding A Python framework for building deep learning models with optimized encoding layers and batch normalization. 2,044