flax

Util to let bridge module work with NNX submodules

#4584

Merged