flax
Replace flax named_call with upstream version jax.named_call
#647
Merged

Loading