Supporting multiple frameworks with TFLite

The machine learning (ML) models you use with TensorFlow Lite can be trained using JAX, PyTorch or TensorFlow and then converted to a TFLite flatbuffer format.

See the following pages for more details:

An overview of the TFLite Converter which is an important component of supporting different frameworks with TFLite is on Model conversion overview.