Add gradient_checkpointing parameter to FlaxWhisperEncoder
Ref https://github.com/huggingface/transformers/pull/23173#discussion_r1188815621

@sanchit-gandhi 
_The documentation is not available anymore as the PR was closed or merged._<|||||>Merging as the failing tests are also failing on main and not related to this PR