[not for land] add repro script for https://github.com/pytorch/torchtitan/pull/808

[not for land] add repro script for https://github.com/pytorch/torchtitan/pull/808 #1838