pytorch

github.com ↗

Tensors and Dynamic neural networks in Python with strong GPU acceleration.