3 Commits

Author	SHA1	Message	Date
sshahrokhi	6f72e71f97	changing the requirements to a cpu torch version that works (#23483)	2023-05-22 12:58:55 -04:00
Suraj Patil	85a4bda4f4	bump flax version (#14343)	2021-11-09 22:15:22 +05:30
Suraj Patil	f1c81d6b92	[Flax] ViT training example (#12300) ... * begin script * clean example, add readme * update readme * remove decay mask * remove masking * update readme & make flake happy	2021-07-05 18:23:03 +05:30