2 Commits

Author	SHA1	Message	Date
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