pytorch

https://github.com/pytorch/pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

https://pytorch.org/