/home/deepak/sources/transformers/examples/flax/language-modeling/run_clm_flax.py