pytorch/pytorch: Tensors and Dynamic neural networks in Python with strong GPU acceleration
Category: ml-dl
Language: Python
Topics: autograd, deep-learning, gpu, machine-learning, neural-network, numpy, tensor
Homepage: https://pytorch.org