flax
9ae3e18f - Change the type signature of nn.compact to transparently pass its input through

Commit

4 years ago

Change the type signature of nn.compact to transparently pass its input through PiperOrigin-RevId: 359460262

References

Author

rchen152

Committer

a-googler

Parents