An simple pytorch implementation of Flash MultiHead Attention

Issues looking for funding

Creatorkyegomez
Stars13
LicenseMIT License
RepositoryGitHub
Websitediscord.gg