PyTorch

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

Zerui Wang
Zerui Wang
Ph.D. Student