Any solution to support llama2 finetune?

[LeiWang1999]

Hi all, I notice that the latest transformers repo doesn't have flax version of llama model, any solutions to support llama finetune?

There exists a previous pr (#923) to support llama model conversion, but looks like it didn't support GQA (llama2).

Thanks for all your suggestions :)