sustcsonglin / flash-linear-rnn Public

Notifications You must be signed in to change notification settings
Fork 1
Star 38

Implementations of various linear RNN layers using pytorch and triton

38 stars 1 fork Branches Tags Activity

Name		Name	Last commit message	Last commit date
Latest commit History 3 Commits
linear_rnn		linear_rnn
tests		tests
.gitignore		.gitignore
README.md		README.md

Repository files navigation

Linear RNN Layer Zoo in Pytorch

Support:

TODO:

S5
Simple Recurrent Unit (SRU)

Requirement

Install triton https://github.com/openai/triton