flax
[nnx] remove VariableState
#4800
Merged

Loading