PyTorch 基金会宣布释出 PyTorch 2.0。PyTorch 是一个开源 Python 机器学习库。新版的变化包括:稳定版 Accelerated Transformers(前称 Better Transformers),torch/compile 作为其主 API,Nvidia 和 AMD GPU 将依靠 OpenAI Triton 深度学习编译器生成高性能代码隐藏底层硬件细节,Metal Performance Shaders 后端在 macOS 平台提供了 GPU 加速的 PyTorch,在 AWS Graviton CPU 上提供更快的推理性能,等等。更多可浏览发布公告。
https://pytorch.org/blog/pytorch-2.0-release/