Swift for TensorFlow: A portable, flexible platform for deep learning
arXiv:2102.13243
Abstract
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.
References in corpus (9)
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications
- EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks
- DyNet: The Dynamic Neural Network Toolkit
- Glow: Graph Lowering Compiler Techniques for Neural Networks
- OpenSpiel: A Framework for Reinforcement Learning in Games
- TensorFlow Eager: A Multi-Stage, Python-Embedded DSL for Machine Learning
- Automatic Full Compilation of Julia Programs and ML Models to Cloud TPUs
- Jelly Bean World: A Testbed for Never-Ending Learning