Swift for TensorFlow is a deep learning platform that scales from mobile devices to clusters of hardware accelerators in data centers. It combines a language-integrated automatic differentiation system and multiple Tensor implementations within a modern ahead-of-time compiled language oriented around mutable value semantics. The resulting platform has been validated through use in over 30 deep learning models and has been employed across data center and mobile applications.
@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}
}