* remove match_shape * ported fixes from #479 to flax * remove unused argument * typo * remove warnings
Attach a file by drag & drop or click to upload
