flashinfer.cake\_rmsnorm\_train.is\_cake\_rmsnorm\_train\_supported =================================================================== .. currentmodule:: flashinfer.cake_rmsnorm_train .. autofunction:: is_cake_rmsnorm_train_supported