flax

63df54ad - Copybara import of the project:

Copybara import of the project: -- 5e75d5b7f0ac9cf0c01bdf7183aa4c1b83dbccfc by Cristian Garcia <cgarcia.e88@gmail.com>: improve submodule binding logic PiperOrigin-RevId: 531595051