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