[WIP] Add Flax diverse group search#24508

yeandy wants to merge 1 commit intohuggingface:mainhuggingface/transformers:mainfrom

yeandy:flax_diverse_beam_searchyeandy/transformers:flax_diverse_beam_searchCopy head branch name to clipboard