3 Commits

Author	SHA1	Message	Date
Suraj Patil	2d42915abe	[examples/flax] add adafactor optimizer (#12544) ... * add adafactor * Update examples/flax/language-modeling/run_mlm_flax.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>	2021-07-07 11:50:30 +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