flax
5d9fd9fb - Merge pull request #3782 from chiamp:nnx_module_subclass

Commit

1 year ago

Merge pull request #3782 from chiamp:nnx_module_subclass PiperOrigin-RevId: 620969972

References

Author

a-googler

Parents