Path	Lines of Code
jax-projects/big_bird/bigbird_flax.py	252
jax-projects/big_bird/evaluate.py	118
jax-projects/big_bird/prepare_natural_questions.py	264
jax-projects/big_bird/sweep_flax.yaml	16
jax-projects/big_bird/train.py	62
jax-projects/dataset-streaming/run_mlm_flax_stream.py	416
jax-projects/hybrid_clip/configuration_hybrid_clip.py	39
jax-projects/hybrid_clip/modeling_hybrid_clip.py	261
jax-projects/hybrid_clip/run_hybrid_clip.py	406
jax-projects/model_parallel/partitions.py	43
jax-projects/model_parallel/run_clm_mp.py	479
jax-projects/wav2vec2/run_wav2vec2_pretrain_flax.py	425