/home/shahu_kareem/transformers/examples/flax/language-modeling/run_mlm_flax.py