flashinfer.cake_mhc_train.is_cake_mhc_train_head_supported

flashinfer.cake_mhc_train.is_cake_mhc_train_head_supported(device: device, hidden: int | None = None) → bool

Whether device (and optionally the hidden size hidden) has the exported head programs.