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