* [Flax] Improve Robustness of Back-Prop Tests * check equality of logits/outputs * make fixup
Attach a file by drag & drop or click to upload
