flax

Add lifted transforms section to Haiku migration guide

#3158

Merged