flax
3d1f55a0 - [nnx] add state summaries for print and display

Commit

1 year ago

[nnx] add state summaries for print and display rollforward #4438 PiperOrigin-RevId: 714046512

References

Author

Cristian Garcia

Committer

a-googler

Parents