flax
c3afa829 - Merge pull request #441 from levskaya:linen_wmt

Commit

5 years ago

Merge pull request #441 from levskaya:linen_wmt PiperOrigin-RevId: 329701094

References

#441 - Update attention layers with simpler flat forms, update wmt example.

Author

a-googler

a-googler

Parents

Loading