Swift for TensorFlow:一个可移植、灵活的深度学习平台
机器学习
2021-03-01 v1 分布式、并行与集群计算
摘要
Swift for TensorFlow 是一个深度学习平台,可从移动设备扩展到数据中心内的硬件加速器集群。它将语言集成的自动微分系统与多种 Tensor 实现结合在一个围绕可变值语义构建的现代提前编译语言之中。该平台已通过在 30 多个深度学习模型中的使用得到验证,并被应用于数据中心和移动端应用中。
引用
@article{arxiv.2102.13243,
title = {Swift for TensorFlow: A portable, flexible platform for deep learning},
author = {Brennan Saeta and Denys Shabalin and Marc Rasi and Brad Larson and Xihui Wu and Parker Schuh and Michelle Casbon and Daniel Zheng and Saleem Abdulrasool and Aleksandr Efremov and Dave Abrahams and Chris Lattner and Richard Wei},
journal= {arXiv preprint arXiv:2102.13243},
year = {2021}
}