flax icon indicating copy to clipboard operation
flax copied to clipboard
verified publisher icon google

Update NNX `State` class docs in `statelib.py`

Open 8bitmp3 opened this issue 1 year ago • 2 comments

8bitmp3 avatar Dec 17 '24 00:12 8bitmp3

May need to review/close because of "[nnx] remove State #4418" https://github.com/google/flax/pull/4418

8bitmp3 avatar Dec 19 '24 06:12 8bitmp3

May need to review/close because of "[nnx] remove State #4418" https://github.com/google/flax/pull/4418

8bitmp3 avatar Dec 19 '24 06:12 8bitmp3