flax
ae208ee7 - Merge pull request #4811 from google:nnx-pytree-modules-2

Commit

260 days ago

Merge pull request #4811 from google:nnx-pytree-modules-2 PiperOrigin-RevId: 786726275

References

Author

a-googler

Parents