flax
fb4f2a45 - Handle setup with transformed methods taking submodules of self.

Commit

4 years ago

Handle setup with transformed methods taking submodules of self. Fixed version of PR #1745 PiperOrigin-RevId: 420035024

Author

levskaya

Committer

a-googler

Parents