/flash-linear-rnn

Implementations of various linear RNN layers using pytorch and triton

Primary LanguagePython

Stargazers