The machine learning (ML) models you use with LiteRT 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.