flashinfer.cake\_mhc\_train.cake\_mhc\_train\_head\_backward ============================================================ .. currentmodule:: flashinfer.cake_mhc_train .. autofunction:: cake_mhc_train_head_backward