Home/AI/PyTorch
P
Website Repository
Capabilities:code
License:BSD-3-Clause
Open sourceLocal runtime

PyTorch

About

The open-source deep learning framework behind most modern AI research and production models. Dynamic computation graphs, GPU acceleration and the torch ecosystem.

Pricing

Free

PyTorch is an open-source machine learning framework with a focus on flexibility and speed. Its eager execution model makes research iteration natural, while TorchScript and torch.compile optimize production.

Key Features

  • Dynamic computation graphs.
  • GPU acceleration with CUDA and MPS.
  • TorchHub for reusable models.
  • torch.compile for up to 2x speedups.
  • Distributed training built in.
  • Ecosystem: torchvision, torchaudio, Hugging Face.

Pricing

PyTorch is free and open source under the BSD-3-Clause license.

Getting Started

pip install torch
import torch
x = torch.rand(3, 3, device="cuda" if torch.cuda.is_available() else "cpu")
y = x @ x.T

Check pytorch.org/get-started for platform-specific installs.