flax
Fix init/apply return annotation
#1020
Merged

Loading