Google DeepMind·· 2022-10-05
DeepMind 发布 AlphaTensor:用强化学习发现高效矩阵乘法算法
Discovering novel algorithms with AlphaTensor
AI 导读
DeepMind 于 2022 年介绍 AlphaTensor,这是首个用于发现新颖、高效且可证明正确算法的 AI 系统,并将矩阵乘法算法搜索转化为单人游戏,以强化学习训练。文章称其发现了多种优于既有算法的方案;针对 Nvidia V100 GPU 和 Google TPU v2 优化的算法,在相同硬件上比常用算法快 10%至 20%。
来源:Google DeepMind · deepmind.google