chore: remove nv-grouped-gemm dependency

docker/common/install_source_wheels.sh

@@ -40,18 +40,14 @@ MAMBA_WHEEL=$(ls $INPUT_WHEEL_DIR/mamba*.whl) || true
 CAUSALCONV1D_WHEEL=$(ls $INPUT_WHEEL_DIR/causal_conv1d*.whl) || true
 [ -z "$CAUSALCONV1D_WHEEL" ] && CAUSALCONV1D_WHEEL=$(bash docker/common/build_causalconv1d.sh --output-wheel-dir $INPUT_WHEEL_DIR | tail -n 1)
 
-GROUPEDGEMM_WHEEL=$(ls $INPUT_WHEEL_DIR/grouped_gemm*.whl) || true
-[ -z "$GROUPEDGEMM_WHEEL" ] && GROUPEDGEMM_WHEEL=$(bash docker/common/build_groupedgemm.sh --output-wheel-dir $INPUT_WHEEL_DIR | tail -n 1)
-
 # Override deps that are already present in the base image
 # only for dev
 if [ "$ENVIRONMENT" = "dev" ]; then
     uv pip install --no-cache-dir --no-deps $TE_WHEEL
 fi
 
-# Install heavy optional deps like mamba, causalconv1d, groupedgemm
+# Install heavy optional deps like mamba, causalconv1d
 uv pip install --no-cache-dir \
     $MAMBA_WHEEL \
     $CAUSALCONV1D_WHEEL \
-    $GROUPEDGEMM_WHEEL \
     "setuptools<80.0.0,>=77.0.0"

pyproject.toml

@@ -100,7 +100,6 @@ dev = [
     "mamba-ssm~=2.2",
     "causal-conv1d~=1.5",
     "flash-linear-attention~=0.4.0",
-    "nv-grouped-gemm~=1.1",
     "megatron-energon[av_decode]~=6.0",
     "av",
     "flashinfer-python~=0.5.0",
@@ -124,7 +123,6 @@ lts = [
     "opentelemetry-api~=1.33.1",
     "mamba-ssm~=2.2",
     "causal-conv1d~=1.5",
-    "nv-grouped-gemm~=1.1",
     "megatron-energon[av_decode]~=6.0",
     "av",
     "flashinfer-python~=0.5.0",
@@ -183,7 +181,6 @@ no_pypi_wheels = ["flash_mla", "emerging_optimizers"]
 default-groups = ["linting", "build", "test"]
 no-build-isolation-package = [
     "causal-conv1d",
-    "nv-grouped-gemm",
     "flash_mla",
     "mamba-ssm",
     "transformer-engine",