Nathancgy/Co-ALIBI

Hardware-aligned Triton kernels of Co-ALIBI (Contextual Attention with Linear Biases)

语言：Python · GitHub 星标：9 · Fork：2

在 GitHub 上查看原仓库