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

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