3 Commits
Author SHA1 Message Date
HowToSD 2168836d13 Fixed the docstring for MHA custom class 2025-04-07 19:37:13 -07:00
HowToSD ec9339a62b Fixed the docstring for MHA custom class 2025-04-07 19:36:30 -07:00
HowToSD 080a23778a Added custom multihead attention code. 2025-04-07 19:32:55 -07:00