flax

Update NNX `Variable` class docs in `variablelib.py`

#4448

Open