jax-projects/dataset-streaming/run_mlm_flax_stream.py
performer/run_mlm_performer.py

jax-projects/model_parallel/run_clm_mp.py
performer/run_mlm_performer.py

jax-projects/hybrid_clip/run_hybrid_clip.py
performer/run_mlm_performer.py

jax-projects/wav2vec2/run_wav2vec2_pretrain_flax.py
performer/run_mlm_performer.py

jax-projects/hybrid_clip/modeling_hybrid_clip.py
performer/modeling_flax_performer.py