Sanchit Gandhi 485bbe79d5 [Flax] Add remat (gradient checkpointing) (#17843) ... * [Flax] Add remat (gradient checkpointing) * fix variable naming in test * flip: checkpoint using a method * fix naming * fix class naming * apply PVP's suggestions from code review * make fix-copies * fix big-bird, electra, roberta * cookie-cutter * fix flax big-bird * move test to common		2022-07-01 18:33:54 +01:00
..
adding_a_missing_tokenization_test	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
adding_a_new_example_script	Add examples telemetry (#17552)	2022-06-07 11:57:52 -04:00
adding_a_new_model	[Flax] Add remat (gradient checkpointing) (#17843)	2022-07-01 18:33:54 +01:00