IBM 如何将 Spyre 打造为原生 PyTorch 设备

导读

IBM 通过 torch-spyre 将 Spyre 接入 PyTorch 的 device、allocator、stream 与编译器抽象，使其成为原生 PyTorch 设备。