Remove torchao nightly version check in mx component (#1872)

Nightly build is no longer required for MXFP8

Author: danielvegamyhre

torchtitan/components/quantization/mx.py

@@ -5,7 +5,6 @@
 # LICENSE file in the root directory of this source tree.
 
 from functools import partial
-from importlib.metadata import version
 from importlib.util import find_spec
 from typing import Any, List
 
@@ -108,14 +107,6 @@ def __init__(self, job_config: JobConfig, parallel_dims: ParallelDims):
             raise ImportError(
                 "torchao is not installed. Please install it to use MXFP8 linear layers."
             )
-        torchao_version = version("torchao")
-
-        # Require latest release or nightly builds for prototype features
-        is_nightly_build = torchao_version.startswith("0.14.0")
-        if not is_nightly_build:
-            raise ImportError(
-                f"torchao version {torchao_version} is too old, please install torchao nightly build and try again"
-            )
 
         # Can be removed if we enable the emulated versions
         assert has_cuda_capability(