flax

8e9f8ae3 - Copybara import of the project:

Copybara import of the project: -- 803cd5d4d0d5aacdc0a1b3fbc5718d32a9209554 by Cristian Garcia <cgarcia.e88@gmail.com>: add HijaxVariable PiperOrigin-RevId: 829630301