Use Float8TrainingOpConfig instead of removed FP8GroupedMMConfig alias

FP8GroupedMMConfig was a temporary BC alias in torchao that has been
removed (pytorch/ao#4069). Use the canonical Float8TrainingOpConfig name.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: Li

torchtitan/components/quantization/float8.py

@@ -274,7 +274,7 @@ def convert(self, model: nn.Module):
         from torchao.quantization.quant_api import quantize_
 
         try:
-            from torchao.prototype.moe_training.config import FP8GroupedMMConfig
+            from torchao.prototype.moe_training.config import Float8TrainingOpConfig
         except ImportError as e:
             raise ImportError(
                 "torchao installation does not have MoE training support. Please install torchao nightly build."
@@ -293,7 +293,7 @@ def moe_module_filter_fn(mod: nn.Module, cur_fqn: str) -> bool:
             model, ["_init_mean", "_init_std"], nn_module_cls=nn.Linear
         )
 
-        config = FP8GroupedMMConfig()
+        config = Float8TrainingOpConfig()
         quantize_(model, config=config, filter_fn=moe_module_filter_fn)
 
         # Re-inject Linear protocol and re-attach attrs