Skip to content

vllm.models.muse_spark.nvidia.ops.cute_dsl

Modules:

  • moe_finalize_rmsnorm –

    CuTeDSL fused MoE finalize with Muse Spark's post-expert RMSNorm.

  • qk_norm –

    CuTeDSL fused QK-norm for Muse Spark attention.