Soumith Chintala<OD>

PyTorch logo



PyTorch is an open-source machine learning library for Python, based on Torch, used for applications such as natural language processing. One can also reuse Python packages such as NumPy, SciPy and Cython to extend PyTorch when needed. PyTorch provides two high-level features: Tensor computation (like NumPy) with strong GPU acceleration Deep neural networks built on a tape-based autodiff system - on GitHub repo Automatic differentiation is done with a tape-based system at both a functional and neural network layer level.

Read more →