--model-type=transformer --distribution-strategy=llm-training --enable-saturate-infinity -O2 --model-type=transformer