philipturner/metal-flash-attention

FlashAttention (Metal Port)

语言：Swift · GitHub 星标：614 · Fork：43

话题：artificial-intelligence · attention-mechanism · high-performance-computing · metal · software-engineering · stable-diffusion · transformer-models

在 GitHub 上查看原仓库