flax
4ed137c5 - [nnx] stabilize unsafe_pytree

Commit
1 year ago
[nnx] stabilize unsafe_pytree
Author
Committer
Parents
Loading