Marc van Zee 4ce6bcc310 Adds Flax BERT finetuning example on GLUE (#11564) ... * Adds Flax BERT finetuning example * fix traced jax tensor type * Use Optax losses and learning schedulers * Add 1GPU training results * merge into master & make style * fix input * del file * Fix bug in loss and add torch runs * finish bert flax fine-tune * Update examples/flax/text-classification/README.md * Update examples/flax/text-classification/run_flax_glue.py * add requirements * finalize * finalize Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> Co-authored-by: Patrick von Platen <patrick@huggingface.co>		2021-05-11 19:02:59 +01:00
..
language-modeling	[FlaxRoberta] Add FlaxRobertaModels & adapt run_mlm_flax.py (#11470)	2021-05-04 19:57:59 +02:00
text-classification	Adds Flax BERT finetuning example on GLUE (#11564)	2021-05-11 19:02:59 +01:00