AI Edge Torch: High Performance Inference of Pytorch Models on Mobile Devices Released today, AI Edge Torch enables support for PyTorch, JAX, Keras, and Tensorflow with TFLite. Source: Google Developers Blog