flax
f411a322 - modify NNX to use id(variable) instead of nnx.Variables as dictionary keys

Commit

1 year ago

modify NNX to use id(variable) instead of nnx.Variables as dictionary keys

References

#4814 - Modify NNX to use id(variable) instead of nnx.Variables as dictionary

Author

divyashreepathihalli

divyashreepathihalli

Parents

Loading