flax

038d512e - Update Haiku migration guide.

Update Haiku migration guide. Added a few clarifications where neccessary and did a rewrite of the Haiku code in the "multiple methods" section to reflect a more canonical pattern (using ``hk.multi_transform``).