add support for hijax Variables in nnx transforms#5065

cgarciae wants to merge 1 commit intohijax-variablegoogle/flax:hijax-variablefrom

hijax-in-graphgoogle/flax:hijax-in-graphCopy head branch name to clipboard