Nightly version check should look for all versions above 0.12.0

[chinmaybaikar]

Bug description

I pulled the torchao repo and built from main. This put my torchao version at 0.14 but when running the torchtitan job I got the following error:

ImportError: torchao version 0.14.0+git8669213bf is too old, please install torchao nightly build and try again
I checked the mx.py and it seems that the code is only checking for v0.13 instead of all versions above v0.12.0.

torchtitan/torchtitan/components/quantization/mx.py

Line 43 in 17ef753

is_nightly_build = torchao_version.startswith("0.13.0")

Versions

torch 2.9.0.dev20250826+cu129
torchdata 0.11.0
torchtitan 0.1.0

[fegin]

cc., @danielvegamyhre