PyTorch

Go to Project Site

PyTorch is a Python package that provides tensor computation (like NumPy) with strong GPU acceleration.