简介
PyTorch 是一个广泛使用的机器学习开源库。在过去几年中,Arm 与我们的合作伙伴一起不断增强 PyTorch 的推理性能。在本博客中,我们将介绍如何利用 Arm 计算库和 KleidiAI 库中的 Kleidi 技术提高 PyTorch 在 Arm Neoverse 上的推理性能。
作者:Ashok Bhat
文章来源:https://community.arm.com/arm-community-blogs/b/infrastructure-solutions-blog/posts/faster-pytorch-inference-using-kleidi-on-arm-neoverse
欢迎大家点赞留言,更多Arm技术文章动态请关注极术社区Arm技术专栏欢迎添加极术小姐姐微信(id:aijishu20)加入技术交流群,请备注研究方向。