Flax Implementation https://github.com/huggingface/transformers/pull/21867 (#2)

29e70d2

nielsr HF staff Shubhamai commited on Mar 26, 2023