flax
0eeab5aa - Update einsum layer for Gemma example

Commit
335 days ago
Update einsum layer for Gemma example PiperOrigin-RevId: 718899289
Author
Committer
Parents
Loading