["--target=trn2", "--model-type=transformer", "--distribution-strategy=llm-training", "--enable-saturate-infinity"]