flax
431291ed - Improve unregistered data detection

Commit
6 hours ago
Improve unregistered data detection Improve the logic to static attributes that contain data types on nnx.Pytree construction. PiperOrigin-RevId: 872945688
Author
Cristian Garcia
Committer
Parents
Loading