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

Why is this technology in trial?

  • Pytorch is easier to learn for Software Engineers than Tensorflow.

  • A lot of R&D these days is happening in Pytorch, and critical building blocks e.g. the Transformers NLP lib show a preference for Pytorch.

  • With regards to Mobile & Ops, Pytorch has been catching up in 2021 with TorchServe and Pytorch Live.

Related technologies