flax
Add `is_initializing` API
#4550
Merged

Loading