Add bloom flax
# What does this PR do?

Adds the Flax BLOOM model, superseding #18022 (where force pushing to Younes' branch after rebase closed the PR branch and created a new one)
_The documentation is not available anymore as the PR was closed or merged._<|||||>Thanks for the clinical review! Merging 👍