flax

6a2b33e2 - [nnx] improve Variable proxy for binary operations