flax

95a9f570 - [flax.linen] Change nn.Module.sow to enable interventions at sown values using intercept_methods.

[flax.linen] Change nn.Module.sow to enable interventions at sown values using intercept_methods. This requires changing the return value (which would need to be used by the calling code), and running the interceptors. PiperOrigin-RevId: 610717214