flax

148f202b - Add flag for variables in nnx.merge