Researcher Edition
Adam Paszke, Sam Gross, Soumith Chintala, Francisco Massa, Adam
Lerer, James Bradbury, Gregory Chanan, Trevor Killeen, Zeming Lin,
Natalia Gimelshein, Alban Desmaison, Andreas Kopf, Edward Yang, Zach
Devito, Martin Raison, Alykhan Tejani, Sasank Chilamkurthy & Team
What is PyTorch?
What is PyTorch?
Ndarray library
with GPU support
automatic differentiation
engine
gradient based
optimization package
Deep Learning
Reinforcement Learning
Numpy-alternative
Utilities
(data loading, etc.)
ndarray library
•
np.ndarray <-> torch.Tensor
•
200+ operations, similar to numpy
•
very fast acceleration on NVIDIA GPUs