flashinfer.cake\_mhc\_train.cake\_mhc\_train\_forward\_prepare ============================================================== .. currentmodule:: flashinfer.cake_mhc_train .. autofunction:: cake_mhc_train_forward_prepare