Released today, AI Edge Torch enables support for PyTorch, JAX, Keras, and Tensorflow with TFLite.

GoogleDevs' platform is a central hub for developers interested in Google technologies, APIs, and developer tools. Through articles, tutorials, and documentation, GoogleDevs offers insights into building applications using Google Cloud Platform, Android, Chrome, and other Google services. Developers can learn about cloud computing, machine learning, and mobile app development with Google's developer tools and platforms.

Google Developers

Google AI Edge Torch is a new offering that provides a direct path from PyTorch to the TensorFlow Lite runtime for high-performance inference of PyTorch models on mobile devices. It offers easy conversion, great model coverage, and CPU performance. The release includes coverage and performance improvements over existing workflows, early adoption by Shopify for on-device background removal, and partnerships with hardware companies for improved performance and coverage. Qualcomm has also released a new TensorFlow Lite delegate that provides significant speedups using DSP and neural processing units.

AI Edge Torch: High Performance Inference of Pytorch Models on Mobile Devices