GitHub repository: yiakwy-xpu-ml-framework-team/NV_grouped_gemm
Description: PyTorch bindings for CUTLASS grouped GEMM for MoE.