Tensors and Dynamic neural networks in Python
with strong GPU acceleration.
PyTorch is a deep learning framework that puts Python first.
We are in an early-release Beta. Expect some adventures.
Source: PyTorch
Source: PyTorch