flax

[nnx] improve Variable proxy for binary operations

#4641

Merged