PyTorch是Meta开发的开源深度学习框架,在2019年后超越了TensorFlow,成为AI研究和工业应用的第一选择。
核心特性:动态计算图(每次前向传播可以不同,调试友好)、Pythonic设计(像写普通Python一样自然)、自动微分(反向传播对开发者透明)。
PyTorch 2.0+:torch.compile JIT编译速度提升2-5倍,多后端支持(CUDA/ROCm/MPS),更好的分布式训练支持。
生态:HuggingFace首选框架,TorchVision/TorchText/TorchAudio领域扩展,FSDP全分片数据并行。
PyTorch的胜利是Python哲学的胜利——好的框架不应该强制开发者改变思维方式。几乎所有主流开源模型都是PyTorch训练的。