WEIHAN LI

icon indicating a website link https://weihanlikk.github.io/

icon location AI for Neuroscience

Results 1 issues of WEIHAN LI

gates (A matrix) with a shape of batch * dim * dim * seqlen

4 comment

Thank you for your excellent work! I was wondering if it’s possible to modify your code to handle a state-space model case where the gates (A matrix) have a more...