flax
1f6b0949 - Flax layers are parametrizable with custom dot_general.

Commit

3 years ago

Flax layers are parametrizable with custom dot_general. PiperOrigin-RevId: 508438464

Author

a-googler

Committer

a-googler

Parents