pytorch

PyTorch is a Python deep learning framework providing GPU-accelerated tensor computation and tape-based autograd for training neural networks in research and production.