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