Feature Request: Flax Whisper
### Feature request

It already has TF and Pytorch support. Would be nice to use it on Flax as-well.

cc @sanchit-gandhi <|||||>Hey @OhadRubin! Thanks to the fantastic work by @andyehrenberg it's nearly complete: https://github.com/huggingface/transformers/pull/20479

Will do a final review tomorrow and then get it merged asap!<|||||>This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the [contributing guidelines](https://github.com/huggingface/transformers/blob/main/CONTRIBUTING.md) are likely to be ignored.