JiHa-Kim / hedgehog-porcupine-linear-attention

Code implementation in PyTorch of the paper "The Hedgehog & the Porcupine: Expressive Linear Attentions with Softmax Mimicry"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

hedgehog-porcupine-linear-attention

Code implementation in PyTorch of the paper "The Hedgehog & the Porcupine: Expressive Linear Attentions with Softmax Mimicry" at https://arxiv.org/abs/2402.04347

About

Code implementation in PyTorch of the paper "The Hedgehog & the Porcupine: Expressive Linear Attentions with Softmax Mimicry"

License:Apache License 2.0


Languages

Language:Python 100.0%