PyTorch: An Imperative Style, High-Performance Deep Learning Library

Deep learning frameworks have often focused on either usability or speed, but not both.

77,638

Paper
Code

Automatic Differentiation in PyTorch

In this article, we describe an automatic differentiation module of PyTorch — a library designed to enable rapid research on machine learning models.

77,638

Paper
Code

Cannot find the paper you are looking for? You can Submit a new open access paper.