flax
7fcc7f6c - use more efficient softmax in dot product attention.

Commit

5 years ago

use more efficient softmax in dot product attention. PiperOrigin-RevId: 319469087

References

Author

levskaya

Committer

a-googler

Parents