Amazon 推出 TorchNeuron:PyTorch 代码原生跑在 Trainium 芯片上
用 PyTorch 的朋友看这个:AWS 让你的代码改一处配置就能跑在 Trainium 芯片上,不用重写,10 月 20 日开源。
Amazon Science 宣布推出 TorchNeuron,让 PyTorch 代码无需重写即可原生运行在 AWS Trainium 芯片上,只需修改一处配置。该项目将在 10 月 20 日的 PyTorchCon 上开源发布。开发者可以把现有 PyTorch 训练和推理流程迁移到 Trainium 硬件,省去改写框架适配层的工作。
Announcing TorchNeuron: @PyTorch code now runs natively on AWS Trainium chips with a single configuration change. No rewrite needed.
Open sourcing at #PyTorchCon on Oct 20: https://t.co/6mz9jVhHTa https://t.co/J3gxbtkkF7