pytorch

Tensors and dynamic neural networks in Python (C++ library)

PyTorch is a Python package that provides two high-level features: * Tensor computation (like NumPy) with strong GPU acceleration * Deep neural networks built on a tape-based autograd system