Update Inference Microbenchmark scripts

[jon-chuang]

maxtext/MaxText/scratch_code/run_inference_microbenchmark.sh

Line 4 in 3519271

 async_checkpointing=false \ 

Is outdated and has arguments missing. Would be nice to have something to run out of the box.