Add a TFLite to JAX conversion example#10

zehuiw wants to merge 4 commits intotensorflow:maintensorflow/codelabs:mainfrom

zehuiw:tflite-jax-colabzehuiw/codelabs:tflite-jax-colabCopy head branch name to clipboard