pytorch/pytorch repository overview

PyTorch is a deep learning framework that puts Python first.

It provides Tensors and Dynamic neural networks in Python with strong GPU acceleration.

http://pytorch.org⁠

Tag summary