You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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?
The text was updated successfully, but these errors were encountered:
billshoo
changed the title
Reference for mathematics of SSMKernelDPLR. _step_state_linear
Reference for mathematics in SSMKernelDPLR._setup_linear and SSMKernelDPLR. _step_state_linear
Apr 21, 2023
billshoo
changed the title
Reference for mathematics in SSMKernelDPLR._setup_linear and SSMKernelDPLR. _step_state_linear
Auto-regressive inference for large N
Apr 21, 2023
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?
The text was updated successfully, but these errors were encountered: