pytorch / pytorch

Python library providing GPU-accelerated tensor computation like NumPy, plus a tape-based autograd system for building deep neural networks.