[Flax] Adapt flax examples to include `push_to_hub`
# What does this PR do?

This PR adapts all Flax examples to automatically push trained checkpoints to the hub
@patil-suraj, I think we should also add a `README.md` for the summarization example (this will probs be used a lot during the sprint).