https://github.com/pytorch/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
https://pytorch.org/