flax
Fix re-export and docs for identity
#4850
Merged

Loading