[runnerup]

Is there a straightforward way to move models/pipelines created in JAX to EdgeTPU (TFLite)?

[SleekEagle]

You can try this https://www.tensorflow.org/lite/examples/jax_conversion/over...