PyTorch - Tensors and Dynamic neural networks in Python with strong GPU acceleration

PyTorch is a deep learning framework that puts Python first. It is a python package that provides Tensor computation (like numpy) with strong GPU acceleration, Deep Neural Networks built on a tape-based autograd system. You can reuse your favorite python packages such as numpy, scipy and Cython to extend PyTorch when needed.

Usually one uses PyTorch either as a replacement for numpy to use the power of GPUs, a deep learning research platform that provides maximum flexibility and speed. PyTorch provides Tensors that can live either on the CPU or the GPU, and accelerate compute by a huge amount.