What is PyTorch?

  • most popular research deep learning framework
  • write fast deep learning code in python (run on GPU/TPU)
  • access pre-built deep learning models (torch hub)
  • whole stack - preprocess data, model data, deploy model in your app/cloud
  • originally designed and used in house by Facebook
  • used by Tesla, Microsoft, OpenAI

*TPU - tensor processing unit

an example PyTorch workflow:

202510231417