PyTorch – Tensors and Dynamic neural networks in Python
It is worth adding that there is a wip branch focused on making PyTorch tensors distributable across machines in a master-workers model: <a href="https://github.com/apaszke/pytorch-dist/" rel="nofollow">https://github.com/apaszke/pytorch-dist/</a>