PyTorch：Python 的 GPU 加速张量与动态神经网络框架

PyTorch 让你像使用 NumPy 一样在 GPU 上操作张量，并通过基于 tape 的 autograd 系统构建可变形的深度网络，适合科研与生产。

· 兼容 CUDA、ROCm、Intel GPU，支持多平台（Linux、Windows、macOS、Jetson）
· 纯 Python 接口，易于扩展；可直接使用 NumPy、SciPy、Cython 等库
· 轻量框架，内置 MKL、cuDNN、NCCL，内存占用低，训练大模型更高效

PyTorch 是深度学习研究与工业落地的首选框架。