Add flax text class colab (#11824)
Add flax text class colab (#11824)

* fix_torch_device_generate_test

* remove @

* add flax glue link
huggingface/transformers,huggingface/transformers,huggingface/pytorch-transformers,huggingface/transformers,huggingface/transformers