挑战英伟达 CUDA:亚马逊开源工具 TorchNeuron 自研 Trainium 芯片轻松运行 PyTorch 代码
← 返回今日版面来源
2 条为了降低AI硬件采用障碍并提供替代计算选择,Amazon宣布开源TorchNeuron,该工具允许现有PyTorch代码在AWS Trainium芯片上直接运行,只需少量代码修改。
亚马逊推出开源工具 TorchNeuron,使 PyTorch 代码可在自研 Trainium 芯片上运行,无需修改或重写。TorchNeuron 提升了人工智能行业的硬件选择自由度,降低了成本并加速迭代。