Update Jax to TFLite example to use Jax2TF
Check out this pull request on&nbsp; <a href="https://app.reviewnb.com/tensorflow/tensorflow/pull/61342"><img align="absmiddle"  alt="ReviewNB" height="28" class="BotMessageButtonImage" src="https://raw.githubusercontent.com/ReviewNB/support/master/images/button_reviewnb.png"/></a> 

 See visual diffs & provide feedback on Jupyter Notebooks. 

---

 <i>Powered by <a href='https://www.reviewnb.com/?utm_source=gh'>ReviewNB</a></i> Hi @alankelly Can you please review this PR ? Thank you! Hi @alankelly Can you please review this PR ? Thank you! Hi @alankelly Can you please review this PR ? Thank you! Hi @alankelly Can you please review this PR ? Thank you! Hi @alankelly Can you please review this PR ? Thank you! Hi @alankelly Can you please review this PR ? Thank you! Hi @renjie-liu Can you please review this PR ? Thank you!

 Hi @renjie-liu Can you please review this PR ? Thank you! Hi @renjie-liu Can you please review this PR ? Thank you! Hi @alankelly Can you please review this PR ? Thank you! Hi @alankelly Can you please review this PR ? Thank you! Hi @alankelly Can you please review this PR ? Thank you! Hi @pkgoogle Can you please resolve conflicts? Thank you! This PR is stale because it has been open for 14 days with no activity. It will be closed if no further activity occurs. Thank you. This PR was closed because it has been inactive for 14 days since being marked as stale. Please reopen if you'd like to work on this further.
The nightly versions says `experimental_from_jax` is deprecated and `Jax2TF` is recommended way of converting the Jax models to TFLite. 

Added an option in the example to use `Jax2TF` for TFLite conversion using concrete functions.

Thanks.