s4 icon indicating copy to clipboard operation
s4 copied to clipboard

Auto-regressive inference for large N

Open billshoo opened this issue 1 year ago • 1 comments

In SSMKernelDPLR._setup_linear and SSMKernelDPLR._step_state_linear, it is documented that:

"Version of the step function that has time O(N) instead of O(N^2) per step, which takes advantage of the DPLR form and bilinear discretization."

Is there any reference, in the series of wonderful S4 papers, for the inference formulas used in this implementation?

Given that it is currently about 2x slower, what's the future plan for this segment of the code?

billshoo avatar Apr 21 '23 09:04 billshoo