FluxML/NNkernels.jl

Flash Attention & friends in pure Julia

语言：Julia · GitHub 星标：20 · Fork：5

话题：amdgpu · cuda · flash-attention · gpgpu · gpu · julia

在 GitHub 上查看原仓库