Nicholas Johnson

software engineer / trainer / AI Guy

What is PyTorch and why is it good?

PyTorch is a framework for machine learning. Made by Meta, it has quickly become the tool of choice for machine learning professionals, taking over from Google’s TensorFlow.

  • Shortcuts to define networks
  • Tensor maths (maths with n-dimensional arrays)
  • Built in Gradient descent!
  • Support for CUDA - the language of NVidia GPUs for massively parallel computation