flax
90e6b606 - Merge pull request #3572 from chiamp:attention_subclass

Commit

2 years ago

Merge pull request #3572 from chiamp:attention_subclass PiperOrigin-RevId: 597129698

References

Author

a-googler

Parents