通过 torch.compile 将 NumPy 代码编译为 C++ 或 CUDA
Quansight 工程师已在 PyTorch 2.1 中实现了通过 torch.compile 跟踪 NumPy 代码的支持。此功能……
Evgeni Burovski, Ralf Gommers 和 Mario Lezcano2023 年 10 月 17 日