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

Commit
162 days ago
modify NNX to use id(variable) instead of nnx.Variables as dictionary keys
Parents
Loading