Flax implementation https://github.com/huggingface/transformers/pull/21472 @sanchit-gandhi (#3)

a2157d3

nielsr HF Staff Shubhamai commited on Mar 10, 2023