flax
0eeab5aa - Update einsum layer for Gemma example

Commit

1 year ago

Update einsum layer for Gemma example PiperOrigin-RevId: 718899289

References

#4498 - Update einsum layer for Gemma example

Author

a-googler

a-googler

Committer

a-googler

a-googler

Parents

Loading