burn_mamba/mamba3/ssd/serial_recalculated/combined_backward.rs
1// TODO: Implement the combined backward for the MIMO-first SSD.
2//
3// The previous implementation used the old Mamba-2 interface (x_bnlhp, dt_discretized_bhnl,
4// b_bnlgr, c_bnlgr, ...). With the MIMO-first refactor, the interface is now:
5// v_bnlrhp [b,n,l,R,H,P], da_bnlh [b,n,l,H], b_bnlrhn [b,n,l,R,H,N], c_bnlrhn, ...
6//
7// Until this is updated, the custom backward is not used.
8// backward.rs delegates to standard autodiff through K2-K5.