4 Commits

Author	SHA1	Message	Date
Suraj Patil	85a4bda4f4	bump flax version (#14343)	2021-11-09 22:15:22 +05:30
Avital Oliver	9b393240a2	Use a released version of optax rather than installing from Git. (#12173) ... Use a released version of optax rather than installing from Git	2021-06-15 16:42:51 +05:30
Patrick von Platen	00440e350f	[Flax MLM] Refactor run mlm with optax (#11745) ... * refactor * update * update * update * refactor run mlm * finalize * refactor more * fix typo * update * finish refactor * modify run mlm * Apply suggestions from code review * Apply suggestions from code review * Apply suggestions from code review * small fixes * upload * upload * finish run mlm script Co-authored-by: Patrick von Platen <patrick@huggingface.co>	2021-05-19 12:00:58 +01:00
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