dependencies/dockerfiles/clean_py_env.Dockerfile,sha256=Rz-vglOzk6aoQQoLsQH8gcRGVTaJom5aYdb6BFfdbRg,2700
dependencies/dockerfiles/clean_py_env_gpu.Dockerfile,sha256=FZZUKq14f9HtsZ8-1Iapx1V6id5ADAAVG94FKqAZhII,3166
dependencies/dockerfiles/maxtext_gpu_dependencies.Dockerfile,sha256=p27BG2e7mc_8Ki8wrgwLC6nP1_ABIs0HmmCdvmTELxQ,3268
dependencies/dockerfiles/maxtext_runner.Dockerfile,sha256=oMgEmCntCXgVFd1r5I3TdK2Y033G21b4XslpS8Ky_NY,611
dependencies/dockerfiles/maxtext_tpu_dependencies.Dockerfile,sha256=KFOMcSCyyCEej_BdJJxUtoxuREdwwrNRcD6ODLIBh9I,3158
dependencies/extra_deps/__init__.py,sha256=3ZO2en3MVpSamgV0n5lAUOAuIKH3pqnakMBXyJAH1QQ,574
dependencies/extra_deps/post_train_base_deps.txt,sha256=apEHlZgcclWVmJRDh_gJxs2CpEppF7phvjRS-jdnxP0,100
dependencies/extra_deps/post_train_github_deps.txt,sha256=pE2PKcZwfFHWCLCh_fC0NomewXwcUvuwPeGBL1kZvNo,232
dependencies/extra_deps/pre_train_github_deps.txt,sha256=sV1RDjptx-NEsGgkDRc8hmMnI62x9Z1In4XjopjXOD4,225
dependencies/extra_deps/tpu_overrides.txt,sha256=G3Zy08YIKXUrWkkNut6zVdtNWRstg5H-u79GJti2VIg,14
dependencies/extra_deps/tpu_post_train_overrides.txt,sha256=WPdY_LsdCjwCQRe8mDJ_Y3yVW9BSthu3i5W0dlvrAUI,61
dependencies/requirements/requirements.txt,sha256=dSbK348kxUuewZwvmnvKW5XU4frbo00DQs7zBlcc9Xk,824
dependencies/requirements/requirements_decoupled_jax_0_7.1.txt,sha256=B5Uux_WEAbx1Xbp-0j9hNMmnBrzncYGwsovb1wE6nrw,758
dependencies/requirements/requirements_docs.txt,sha256=5f01Fw6o4FV5dRhb-RELZ-507zufma5PsDjikqT3AWg,180
dependencies/requirements/base_requirements/cuda12-requirements.txt,sha256=0IbiyqjUMX7jbTaUlqf_h3ZESaTjeN4RZTiLrOBsJ2A,44
dependencies/requirements/base_requirements/requirements.txt,sha256=qLz5FCZ8Enm5-_8ipd8XU1owUBOCXABH8CoLjRYLpT8,622
dependencies/requirements/base_requirements/runner-requirements.txt,sha256=MpoJLyt_GcqRiyLrk-hsn7zSlaYXRIl0o9gJ32mMMM4,71
dependencies/requirements/base_requirements/tpu-post-train-requirements.txt,sha256=_aPefsDngqDzINdYJHU-UvD-J-NsIUAXBCbgt4Jspqs,514
dependencies/requirements/base_requirements/tpu-requirements.txt,sha256=aQdQVMbTv14BBLvF1yiV84DwUCFqdu6JeNxPv9Yvleg,20
dependencies/requirements/generated_requirements/cuda12-requirements.txt,sha256=rdcR6DLpAFZe6lPE4zz9DEnzwN10StTkXnpcpncli64,5900
dependencies/requirements/generated_requirements/runner-requirements.txt,sha256=OJ49BjQSadrzg3bvO1aTVBZ370djl7MCf2ZlpfVubuM,2828
dependencies/requirements/generated_requirements/tpu-post-train-requirements.txt,sha256=ANUB-ygi6i3hZaPajVVdNU8c-YRRFSWLihQVNvMI0ww,7993
dependencies/requirements/generated_requirements/tpu-requirements.txt,sha256=EjVRUxz4w_jgympeUrLbsVbnWF9dttfdUNCCIDsusgE,5021
dependencies/scripts/build_maxtext_docker_image.py,sha256=6FQUAv2gkc6SEzNyUOElwTKqKYcvShB7xxo8ZFh1M0I,1170
dependencies/scripts/docker_build_dependency_image.sh,sha256=aYOkYf2qklJiCywTL2EAg3e2mh2cyVPFf9xu2UKhmWY,4560
dependencies/scripts/docker_upload_runner.sh,sha256=18-FC6SSnkQRt0UNIvNxfiyDKqj5TyRixTKnVUM98o0,4526
dependencies/scripts/generate_requirements.sh,sha256=lGo-KWjy3JWAK5lVcnbl8_nuLX9Hw-nZFzoYB45BHQY,4320
dependencies/scripts/install_post_train_extra_deps.py,sha256=tBOI3H4fVq1TGuKMKyLoB6vs-z-62MsEoZ5pt-hhrJI,3510
dependencies/scripts/install_pre_train_extra_deps.py,sha256=RxunhPNG0pSNWBTNHPA0sr7BUl4U4yL7VyBUmlEzz80,2814
dependencies/scripts/preflight.sh,sha256=z1w2Hphov5X5zgbkxBn9F_6elWQDWJhKSp0Nxkkj4n0,908
dependencies/scripts/rto_setup.sh,sha256=MFyEVSxSzsySVIcbCY9U5W4otnJrwYK_MYdgpgy1yrc,1524
dependencies/scripts/setup.sh,sha256=9E5pVwNuW2BLwhGzKtISanZ7XrgWS6LtMMWgbCecwWc,14091
dependencies/scripts/setup_gcsfuse.sh,sha256=ZDeRzo6POsPHdFcAIECflL3qJXM_cQmsUCa3N4M5eAk,2382
dependencies/scripts/upload_maxtext_docker_image.py,sha256=237gMGAW0i2h3I3UPEFHBetORg_1LrdSr10QSjCr1W4,1152
maxtext/__init__.py,sha256=UF7NCf7hyCHp0eq7U1pN2Z-My9h0YmBVPH08ESOjH-g,1815
maxtext/assets/tokenizers/tokenizer.default,sha256=UDbNXuaw_4jTSRw4u2y5Zl4ItgdzzIHoZc6WRA5bLRU,811940
maxtext/assets/tokenizers/tokenizer.gemma,sha256=YaexRzkMZFhdbDVD3W_GNpBsmvOGWlVI8n8xruHUyOI,4241003
maxtext/assets/tokenizers/tokenizer.gemma3,sha256=qW36HClpd70iUXYgKzBa6ynz9JzH1UsVvTHpmPkblzY,4689072
maxtext/assets/tokenizers/tokenizer.llama2,sha256=nlVq_UQhO2vRviuFDru9mPVIFDeoAhr69Y7n-xgY00c,499723
maxtext/assets/tokenizers/tokenizer.mistral-v1,sha256=2t_VbXZnFcYdLveApSWrQ7jm2k3mhlvaPZX9714TQFU,493443
maxtext/assets/tokenizers/tokenizer.mistral-v3,sha256=N_ADdN6khljuj10PIYlbm8VcsBA5OWB8gYW_0cbKH4k,587404
maxtext/assets/tokenizers/tokenizer_llama3.tiktoken,sha256=gunTGXnpKrkpzVREQPEp2ezXl7aeMn-A8X4cUNVVG1U,2183982
maxtext/assets/tokenizers/qwen3-tokenizer/tokenizer.json,sha256=rrEzB6cazY_oGGHZStVKtonfdzMYgJ7tPL55S0SS2uQ,11422654
maxtext/assets/tokenizers/qwen3-tokenizer/tokenizer_config.json,sha256=1dCfB7SMMIbFCLMNHJEUvRGJFFt06YKiZTUMkjrNgQE,9732
maxtext/checkpoint_conversion/README.md,sha256=a1-t2XrJRavVXhml3CBs0q74IdKiUpRfN4I5hDcpkrE,315
maxtext/checkpoint_conversion/__init__.py,sha256=otX-EDUVsL8kHS275NFH8VnNE-7RDyQmKir2uN1UyGg,581
maxtext/checkpoint_conversion/compare_hf_ckpt.py,sha256=aNslG5jvnlPEJitQ5fTnrxYNMvv7dqFIDxkTiY0O6AM,8680
maxtext/checkpoint_conversion/load_and_quantize_checkpoint.py,sha256=z3Jb2mGeeGjhXz8lhng9AAZwylEinEA3r0jGh3ZJeTg,1681
maxtext/checkpoint_conversion/reshard_checkpoint.py,sha256=Ow2LchfV4crIdAVdNIWNhp2caeh6U8J1Upu7tmXRwDA,6324
maxtext/checkpoint_conversion/to_huggingface.py,sha256=ACJm3Yc6JCE4vJ4ZiZwQDXBhDJF0VgPTDV8pOAHh7O0,15108
maxtext/checkpoint_conversion/to_maxtext.py,sha256=TiOi-OGc7KYZllTjIo683CoZxVO2wbwCVe_UlZ9DJAk,35806
maxtext/checkpoint_conversion/examples/convert_gemma2_to_hf.sh,sha256=FeR1boCp9Q4PjsNuxeXQGEfvhIrml9cY2Sl5dl3QMQs,2940
maxtext/checkpoint_conversion/examples/convert_gemma2_to_mt.sh,sha256=98xQcKeK4hbSpGybCkhNd_xy94mYZMxCB46JhK2dUH0,2462
maxtext/checkpoint_conversion/examples/convert_gemma3_to_hf.sh,sha256=9CtcKsG1Abe_8Vl2dVe50XEPBM-qn3Ri2-yzcyVeqEw,2968
maxtext/checkpoint_conversion/standalone_scripts/convert_deepseek_family_ckpt.py,sha256=5yNn9yxU_7HDxM26DC7Z4Yl3ayPyuD6blbkXZ_ARZM4,42438
maxtext/checkpoint_conversion/standalone_scripts/convert_deepseek_family_unscanned_ckpt.py,sha256=TJQvReSLkeMIGXhBj7toz6HIPFp_WO-usZSyVSgbL5I,17118
maxtext/checkpoint_conversion/standalone_scripts/convert_gemma2_chkpt.py,sha256=hyHHeUDFcs_x307fl5RjFvRhTdFsLDETGEi81qI9i04,12757
maxtext/checkpoint_conversion/standalone_scripts/convert_gemma3_chkpt.py,sha256=t8ND4nk1W8Z_MszVWsLOEtL78IQ7dXgx-cw8lXChuVc,10236
maxtext/checkpoint_conversion/standalone_scripts/convert_gemma_chkpt.py,sha256=k_zc6lC4pcSNvBp8ms9Jy1NhOaDIzgd6pfVSDj6a9Hc,7163
maxtext/checkpoint_conversion/standalone_scripts/convert_gpt3_ckpt_from_paxml.py,sha256=LeBUrnTldM6uljaQgGNph8tOHZvqYmfbJdPNkh9izEk,12727
maxtext/checkpoint_conversion/standalone_scripts/convert_gpt_oss_ckpt.py,sha256=gmz7nNZPTQ6PlQr5nYUPuiVEVqqzdiKFNtusR6_Oy98,17951
maxtext/checkpoint_conversion/standalone_scripts/convert_gpt_oss_unscanned_ckpt.py,sha256=-3qRdursvDUi5ovHj3vVejg01xG-nfiZyaSCt68bLtM,15250
maxtext/checkpoint_conversion/standalone_scripts/convert_qwen3_moe.py,sha256=w9_oPE5BhzFKDsoBvRexUEuM8V2XeOBSLqsRUAvb50g,13528
maxtext/checkpoint_conversion/standalone_scripts/convert_qwen3_next_scanned.py,sha256=9BfdyMnMoXRJ2eZBFUvPPvtqPfzT_axS4I9O5S_L30c,21667
maxtext/checkpoint_conversion/standalone_scripts/convert_qwen3_next_unscanned.py,sha256=ehCvZ2O8CS4OW8LcQcb78N3PDxGKM49gox2n_91KkTc,14088
maxtext/checkpoint_conversion/standalone_scripts/deepseek_fp8_to_bf16.py,sha256=2a_VSB6o-akC0uYh7kNCjqlzS3K4NxbawFHdDPBlMeI,6436
maxtext/checkpoint_conversion/standalone_scripts/dequantize_mxfp4.py,sha256=RoCBBfsf_HDgwmODLiUP59H8us0aGK8f4-sXzFxxNy4,7069
maxtext/checkpoint_conversion/standalone_scripts/llama4_ckpt_unscanned.py,sha256=46xtdslz3EXXn_68HJXLDfNaGUKvHXSQriVi3YnFNdQ,43827
maxtext/checkpoint_conversion/standalone_scripts/llama_ckpt_conversion_inference_only.py,sha256=cyMpYTpIeH1tahAI3Uw6xHWp1g_IFERHgDbcDXEb5Rw,12209
maxtext/checkpoint_conversion/standalone_scripts/llama_mistral_mixtral_orbax_to_hf.py,sha256=7G0HFOX4c96K2F7cSts2Y9umeH-wTm_rTMocPSWKhGU,10533
maxtext/checkpoint_conversion/standalone_scripts/llama_or_mistral_ckpt.py,sha256=rlMN683wc_b28J5dKg09eULtrddWGeHg7WZiI-z10gI,81109
maxtext/checkpoint_conversion/utils/BUILD,sha256=TxcUTLKcDOf81XnjWwq6hi2yOjU0UDTP-29iZ2S51fI,1013
maxtext/checkpoint_conversion/utils/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/checkpoint_conversion/utils/hf_model_configs.py,sha256=L2aijeH7Tvpjf3_C2KsODx5WnH_dISzBv8shGM-4oZE,35923
maxtext/checkpoint_conversion/utils/hf_shape.py,sha256=gfnFuS1o69FsCpUH2iwjMC4Hu9mvFwfQVxjwDabmFdU,41084
maxtext/checkpoint_conversion/utils/hf_utils.py,sha256=TcwALFP7YqStfBHw1areoifc09rfrHIh216Ok3_eCus,8432
maxtext/checkpoint_conversion/utils/param_mapping.py,sha256=Uip8JtNsYxB2n0IYi_R5g9rW1zyUZprEOjWf3JyksN8,144094
maxtext/checkpoint_conversion/utils/utils.py,sha256=OYPPK0JUtcnRKxqWD_y7RxQRsUmqDwsrg5OMVtSnEVg,38743
maxtext/common/checkpointing.py,sha256=62BqABmF4rCQzDKigG1MJ_LhnxjfQb16CS5hqNdGM2Y,37491
maxtext/common/common_types.py,sha256=8iPuXRXymmf_Sw-swMxqaFLUOhpXJ8w3LD1unMSU9dY,3913
maxtext/common/data_loader.py,sha256=-2Wz-SPAkyBtCzx1W43eaw4p1kr7jYkFZBqpyBXehbU,6656
maxtext/common/gcloud_stub.py,sha256=BiApzXnrY_pSLsIVR0stpMGIaPTeGZiA4rr8FP9HmwI,20917
maxtext/common/gcp_workload_monitor.py,sha256=8SDgFgTAFuUGRRvYE_kXfrcbFXmj8rQKlRljFpHjNZo,9109
maxtext/common/goodput.py,sha256=Kr4Bl-HFM4SoG-Cpi_dNgM3qbrJyo15HFu8V0NaxNbA,4629
maxtext/common/managed_mldiagnostics.py,sha256=QmBv4Edeop4OR0Hr1Sj29O2HrfcuBPZkznHNiMUWDJ0,2611
maxtext/common/metric_logger.py,sha256=0nNeEi4i9uYm_G5QFHWfMUA4diJJ7Rl_f0o19rUEYrI,19052
maxtext/common/profiler.py,sha256=Ohl--PNuaHcIpAd6__vf9ClVcnEXDXZfv_ynsuxvngY,7337
maxtext/common/vertex_tensorboard.py,sha256=KULpw-MR-yllPB9nECp4j-fwtlEWHHux6jSUFmQF2Bw,5267
maxtext/configs/README.md,sha256=g4h8Lm82zgi1jyoWUFzeIUzU7y8LeYqeQidjIN-v_DI,32160
maxtext/configs/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/configs/base.yml,sha256=afHBOw1rMahuAk1cM1XJCqpb_n-Y0gBmzhq2g3QJ8Do,67420
maxtext/configs/decoupled_base_test.yml,sha256=bKxf9SvvG2P9ZJ-FKQaSBPFqS_fgWWdOYncJAl7Qub0,2425
maxtext/configs/pyconfig.py,sha256=TyZsgGjd-WVU1JYzkrBvSk3-bFx5-ynRyZ1i-biB-uo,19817
maxtext/configs/pyconfig_deprecated.py,sha256=AdgKnvS3czPSRgV7EeW_ZEl1x0FhmSp_peYR6B7oNVM,62958
maxtext/configs/types.py,sha256=hn78EpnpWHL1bZjzXlk_bfsWc98PLDtAIIH8ihLSavY,146240
maxtext/configs/custom_mesh_and_rule/cp-as-ep.yml,sha256=BdWTiXShT0aUDs665OIGFQhidPE4IPYf8YwXSsiQBVA,4021
maxtext/configs/custom_mesh_and_rule/ep-as-cp.yml,sha256=pdF2A0tNZMAqnxIiGyp544HijAFqoKGWoTWpW9YX4KA,3878
maxtext/configs/custom_mesh_and_rule/pipeline-large-moe.yml,sha256=c02-26NpnV1xHRTsF6zgMSJuz8V40iPveRvQuZ13saQ,3993
maxtext/configs/custom_mesh_and_rule/pure-fsdp.yml,sha256=kjwrb8yORqm0-vH1tT4P5PbP6oDRHjHxpOklifmHzDA,1585
maxtext/configs/experimental/1024b.sh,sha256=pOMosAY4_wuoJFJNjjdx9sAlEfJITSg8DW0quPdGC10,1504
maxtext/configs/experimental/128b.sh,sha256=RdcMajG4EVzoymXnII2CepzuT1Xqrdc1uzTbex3sRXQ,1504
maxtext/configs/experimental/256b.sh,sha256=ogLS7qxtv1Am4cado-Gk-llC0dz6lqR0bXtcfdWDirk,1503
maxtext/configs/experimental/32b.sh,sha256=rJFOol0y81SIEzM5ljIPGlJnL0T_AcS0rsVldHzSB5Q,1501
maxtext/configs/experimental/512b.sh,sha256=LFKcHVpHYxAHPc1iAqflAUqKS-SNhT41ROa_04ObErA,1501
maxtext/configs/experimental/64b.sh,sha256=Gl78SlZ7f9OTZnXZl_qNCySnsZ98BhLfrDjCzdzhlco,1501
maxtext/configs/gpu/gpu_smoke_test.yml,sha256=jtYziEIkCr-qFGL_TbJWbJtEpOM-ZcW5b7IbIM3xiuk,294
maxtext/configs/gpu/a3/llama_2_7b/16vm.sh,sha256=3JdPChW10-oxhR_ZouVJ9SrAK45IdOMFlHRRxp87St0,1706
maxtext/configs/gpu/a3/llama_2_7b/1vm.sh,sha256=XKsDO1ZnXcTyVg_6m92aNNofo9G05EehIStXYIPL4CE,1681
maxtext/configs/gpu/a3/llama_2_7b/2vm.sh,sha256=81CSIfzwkQS7XZNccc9cYBdyq8cS3jXb4vr59eGgZDc,1650
maxtext/configs/gpu/a3/llama_2_7b/4vm.sh,sha256=E67XWZ7gZIQJJwZkIC7HEwNLgAuMbhYTB0lHG1psJ4g,1650
maxtext/configs/gpu/a3/llama_2_7b/8vm.sh,sha256=NBdNW3SQ_fiyagmUoGv6kv2s2MiKoajvbuKKfa4_FF4,1651
maxtext/configs/gpu/a3/llama_2_7b/README.md,sha256=Vu3lkSSbtK_Yt7NCitOMdDfPEM0xJfva5LaooGeJEUM,1042
maxtext/configs/gpu/a3/llama_3.1_405b/128vm.sh,sha256=Ssj0fLp0fQ96-ZmoyMVx67KFx0Ei_Z4prMBHgrlA3aQ,2366
maxtext/configs/gpu/models/llama2_70b.yml,sha256=RoUcizKa1aySwKMinwGSJfbAm3JxFz5dVa4TCrHUkEo,442
maxtext/configs/gpu/models/llama2_7b.yml,sha256=BWUK2uyJCvBgJbxXtiv2FlXlIz-ZuBxJnJkgWIFvy6A,362
maxtext/configs/gpu/models/llama3.1_405b.yml,sha256=OzMIQoFfXcamyi-Nu9PWnxns51z98IOrcCFPW4hkaSk,446
maxtext/configs/gpu/models/llama3_70b.yml,sha256=pcETus-9WRIt0Xt_3mnUvSBjvS4OZFlJien_PCBeXRc,1052
maxtext/configs/gpu/models/llama3_8b.yml,sha256=C1a-UA_m0RIpGIVafNaODox3r2FlWXEPidGTx0RFz-Q,1067
maxtext/configs/gpu/models/mixtral_8x1b.yml,sha256=4LANilw7nDpnydrTQC-_8_nurQmbQ__Yo-9zkhjcG2g,1144
maxtext/configs/gpu/models/mixtral_8x2b.yml,sha256=1maDNPKQknmt7zZpmsH0A7jJIh3MbGA7FRsZiIIU_4Q,1145
maxtext/configs/gpu/models/mixtral_8x7b.yml,sha256=0aqx18FNDR7vMMIbjR5ZaPvc54PjUcks9XD8Eu4bawI,1221
maxtext/configs/inference/inference.yml,sha256=dcQXYh3Tu3HYmOZC5PnCQj9MB8x-3t1trL9I-Y-sc0U,5033
maxtext/configs/inference/inference_jetstream.yml,sha256=xVnFEvb4U30YXxe8_qYLzPKzLQy8ZvrEmBrO5W6ll1E,103
maxtext/configs/inference/vllm.yml,sha256=IjzAm44GsJtpXOTh-hgtHuI5jaJWO3-s7ZbMsAugbMM,5397
maxtext/configs/inference/multihost/disaggregation/llama3_405b_v6e-16-16.yml,sha256=hx0zRYle2UqjuYe1ZP4yacwqMo7RVdpkoIO8KDYTvww,1753
maxtext/configs/inference/multihost/interleaved/llama2_70b_v5e-16.yml,sha256=kD9XYQcyaFS8awGSR06vRDzyoxJVbbR8-Mmqc1rCa1w,1932
maxtext/configs/inference/multihost/interleaved/llama3_405b_v5e-64.yml,sha256=1s6c4m8Sw3GhYBVHW_lsexEnJ59wIzV6Uz-MHdvmKEE,1805
maxtext/configs/inference/multihost/interleaved/llama3_70b_v5e-16.yml,sha256=br7OH5edxaqB2kfGfycI8cXjGI24hnFezMi97Aut92A,1983
maxtext/configs/models/deepseek-custom.yml,sha256=6lwKDZctikJoyqzMOcwkFL1zEc2OQAOVZqYecQLjpJA,2888
maxtext/configs/models/deepseek2-16b.yml,sha256=42HFfnXbWcZfPbpMXMlt_UCzbSIAkWVcjfmu03B3Ulw,1482
maxtext/configs/models/deepseek2-236b.yml,sha256=Skvr7FRNJwGYnRTCYyFNwfGcTvzIz8aK4TXVCGZbMOs,1556
maxtext/configs/models/deepseek3-671b-2dfsdp.yml,sha256=CugfUpEkzA51DdoFmoVfmqOv70TaW9TGcyTujapoXgo,3128
maxtext/configs/models/deepseek3-671b-batchsplit.yml,sha256=ezpClARezHhm7kImWw5cFJenhT8y6UynLPMEaqdhr-Q,2787
maxtext/configs/models/deepseek3-671b.yml,sha256=yOk3_M4vo1__SD-t-K2lAAuAdsRgRQS1vJym_wL3iyg,1619
maxtext/configs/models/deepseek3-test.yml,sha256=1gtOOiWzzzq1vVgc9r8z_rME95nnohUkIMm4CZ0MozQ,1515
maxtext/configs/models/deepseek3-tiny.yml,sha256=ynRgWMrIpDYqfGX9RXj8O1foG0JAnp9pptS1pmfA128,1399
maxtext/configs/models/deepseek3.2-671b.yml,sha256=NXOu08OYbZDFQ2cka8uveYN9QSIEogrwssL--NYhJ-I,1670
maxtext/configs/models/gemma-2b.yml,sha256=Nr0nYgZmkcJR5TiaPOpC3m8chK3qF1XmsmOOL-Am0dY,876
maxtext/configs/models/gemma-7b.yml,sha256=RDF51SXIeO97RwLxjvYPm0JIPufTxhthpIwI_k_98K8,878
maxtext/configs/models/gemma2-27b.yml,sha256=JfWY-8FQbaBhzpAdmaXjR7xxuipnl-8AQGtg1o-uWR4,1117
maxtext/configs/models/gemma2-2b.yml,sha256=lxbeGdq75iiI4hgeJsJWWonjLfJUhOcIjaMS-oLTGy8,1112
maxtext/configs/models/gemma2-9b.yml,sha256=VuE_Nn1LyGEXy4T5kD2Ek6Z0qR9Z4DpjKZGKyD_PA-A,1115
maxtext/configs/models/gemma3-12b.yml,sha256=f2y64stDAlzfokrxAuPW0yj_UHAMpPnPux7ocrmAQB8,1355
maxtext/configs/models/gemma3-27b.yml,sha256=mYaT247kS40tpDxKfOSU3K9RPMO20yW8P191-XzPCic,1356
maxtext/configs/models/gemma3-4b.yml,sha256=r9lTn0jaVLthmIEZuyNLFEkcdw5XJyvj8R9HCT9Q6jA,1353
maxtext/configs/models/gemma4-26b.yml,sha256=nIvVJFVQmnwyacQoQpD3f481eCVJo3546fghgQTtZmM,1782
maxtext/configs/models/gemma4-31b.yml,sha256=e6x6jFE7mOjN-rn_Cp7xFeUchgkk392NvfasPrfWE2I,1661
maxtext/configs/models/gpt-oss-120b.yml,sha256=nEbdZYNBWmIieErjZbdB16xr5j00vmIdaqG-2UdSz20,1502
maxtext/configs/models/gpt-oss-20b.yml,sha256=YXrw_jYPR1_hb_XA2tey0aF98OFL1a-cj3ABoPgzi1Y,1499
maxtext/configs/models/gpt3-175b.yml,sha256=cH_OE2AGGUX3wJvVL0FHNf4Xmd2YNUGcf1U6AHJvG2Q,1321
maxtext/configs/models/gpt3-22b.yml,sha256=mpzAe-z_shkSKDGMkuGr82Z3VvrNfi6u94Ow4br6Go0,1161
maxtext/configs/models/gpt3-52k.yml,sha256=bEH1hnt--FsUFxrLhlEc-TfSCObMX3cIkHzZrnrJLSM,1258
maxtext/configs/models/gpt3-6b.yml,sha256=14JDd5gO2H0vWDaud6kneKuHfZHZT0_j5la-kuptvRQ,1160
maxtext/configs/models/kimi-k2-1t.yml,sha256=fU5o-XPLeRz3Tk-7oDIG3gF4vI9nEpcD92kqQKFdrRw,1530
maxtext/configs/models/llama2-13b.yml,sha256=egsgH8S-tgc1ReqSZI-pM9LSSVK3WbMamqI1dDl7YgY,943
maxtext/configs/models/llama2-70b.yml,sha256=3NnyUoIjKip_BAi579zzlwHOdDvJNYWhoGk9VVHmpn0,919
maxtext/configs/models/llama2-7b.yml,sha256=7OQHoDxG-EPKKyS2iAruO-akUSXPNikr9vXo4OBqTMs,942
maxtext/configs/models/llama3-405b.yml,sha256=5vUFC9H1Dd1AwbH93Jre_2xPy_ghlxt2Aha53dKmpgM,977
maxtext/configs/models/llama3-70b.yml,sha256=ukvOa3txB5EmPazJFb07ndp2qPXd4aBM88UPewFHB8Y,972
maxtext/configs/models/llama3-8b.yml,sha256=YLWlzMhSeWJWkmVjD0IEPp_HkKYV6U5xdH-dEQ9Stk8,971
maxtext/configs/models/llama3.1-405b.yml,sha256=4Q-LQwsxEqLhd1cftHMfD96faPrp2j-N0T_H9X0NgGs,978
maxtext/configs/models/llama3.1-70b.yml,sha256=6uUSfoxUrxH0JfsrK19o5xGaLoDa4M6b2tzdLPAS66s,974
maxtext/configs/models/llama3.1-8b.yml,sha256=3xKJbRbp17l6dtSSLh3QhM42tvdu6n3gbNWf5amEhf8,973
maxtext/configs/models/llama3.3-70b.yml,sha256=Mkg0XEDy8DvT2ChujvTmLcu4fytN50wvKN8RUrJ70uU,974
maxtext/configs/models/llama4-17b-128e.yml,sha256=5P7AXKOuzW0wgGVk1vrSoS8tmYYAeTvhU3HcgsmHdkk,1765
maxtext/configs/models/llama4-17b-16e.yml,sha256=FMDRXQN-qfw-1Ci0IgfyVYSx9c89IRR2hyn7yV9Zmdg,1799
maxtext/configs/models/mistral-7b.yml,sha256=CHrvrAmDN7L4MRFXmrKPd0ba8fd7WnwluoD6q9onnXk,934
maxtext/configs/models/mixtral-8x22b.yml,sha256=dkIwU98YneWrKja_g2zetRsnVtdX_mNElGzuq-MgZaw,1058
maxtext/configs/models/mixtral-8x7b.yml,sha256=Lveljh4dVIRcIwnlkykCIlYZuLK8rQICKuf_o9g9Hpg,1057
maxtext/configs/models/olmo3-32b.yml,sha256=Px8ZwZsHp4ygBbou5ygRMPNtLZXYjmFMNaF4tN7JmC0,1731
maxtext/configs/models/olmo3-7b-pt.yml,sha256=-JrLyQpDeB2AfvPazJ6sOZgmey1gpmWbLEMUwZM5_wk,1875
maxtext/configs/models/olmo3-7b.yml,sha256=bXHTlz19m0o-vo9XkyKw2Lo5N2lg9IJlQmCOH8KOfgg,1715
maxtext/configs/models/qwen2.5-1.5b.yml,sha256=_zz1gbjfC84N5YIky4EKC7gjBBB0MPV1RVmLOAf6JD0,1099
maxtext/configs/models/qwen2.5-14b.yml,sha256=mR_1sg9vgGUP9JNc81MvWP7da8P6OF1rGuACjnAqQo0,1100
maxtext/configs/models/qwen2.5-7b.yml,sha256=HvpAJaeMU3_2o7MD56LeVFyd9SlQd7MBeBzTZREOO4Y,1097
maxtext/configs/models/qwen3-0.6b.yml,sha256=YeKn11cvIoa6u2NyNEtb_TVqouzRazwPutN_VZlGXz0,1118
maxtext/configs/models/qwen3-1.7b-base.yml,sha256=V8g2qWwLVt-PGUJdy91oWfZIw9_GJwrYGWoBKtKX40k,1123
maxtext/configs/models/qwen3-1.7b.yml,sha256=kuaDWpPYTwZZoGByEhpkhsnBMvLxH_zeMgHR37oMRLo,1118
maxtext/configs/models/qwen3-14b-base.yml,sha256=KO7g9ReijIVYTuSqLqSCopqVvEfhCHGQ_2mm0uMMfz4,1107
maxtext/configs/models/qwen3-14b.yml,sha256=ncoKpvffiM2GoGoZXRSpXTX5auHCgIwiUavLCS4TQKw,1102
maxtext/configs/models/qwen3-235b-a22b.yml,sha256=_LkQFnusYfUJXLsHQyhpFE7ELfSEs0QEl725fnXvYdk,1143
maxtext/configs/models/qwen3-30b-a3b-base.yml,sha256=1_qxHBtI3y6EFzbDg_9KDdlJS64e2gFflCVpyuqi0nA,1115
maxtext/configs/models/qwen3-30b-a3b.yml,sha256=Hji4wdfowhYdEEHiyYUuFsRR7frW6-duzyX9XiwUEu4,1110
maxtext/configs/models/qwen3-32b.yml,sha256=ei39Rfekj25ahXbjsaTcSb2Ow1qVGVsAPEyl6-55rp8,1102
maxtext/configs/models/qwen3-480b-a35b.yml,sha256=jo5XNsb0aPDSdrxiKbni7lcYtlZad0UfkmO50zk951g,1129
maxtext/configs/models/qwen3-4b-base.yml,sha256=ortpc8aTdgkwHtprWEvzPBaaS2JBtfU6Ep5IXgDTIrc,1121
maxtext/configs/models/qwen3-4b-thinking-2507.yml,sha256=3vInEdby_JHJxVk1iaxBE1FE9E0mQkai1w_-wOHgJhw,1100
maxtext/configs/models/qwen3-4b.yml,sha256=QhjTz2BMQFQGPGDqYCDVgZ0gGNi-7m5l28C2UGHdbxs,1116
maxtext/configs/models/qwen3-8b-base.yml,sha256=k54FIViP4r_JLYaj568Aj1YhWhyHpbb6LgOtf9wdbOk,1153
maxtext/configs/models/qwen3-8b.yml,sha256=yfPsynvPYQRqsfppg2yz9QpdLboLuWIHcpjcdC94z5E,1148
maxtext/configs/models/qwen3-custom-30b-a3b.yml,sha256=BtpfrWewWlsLZ-SLzLTQV5PwbmFqLt7-xldXCQJmzLA,1185
maxtext/configs/models/qwen3-next-80b-a3b.yml,sha256=pEa6ZSrjc-yoXRWaOaVZNIkAguWuy9sNo_p9ZXIJhOY,1497
maxtext/configs/models/qwen3-omni-30b-a3b.yml,sha256=TigTn69AhO5dJcxCQME03PK9naUwH6yqiYvou9eF-4U,2575
maxtext/configs/models/qwen3.5-397b-a17b.yml,sha256=-mfrEZFz6Egosw9_MZ-6jZaNfxI9N-kfSkDkTb9anT4,1436
maxtext/configs/post_train/distillation-sft.yml,sha256=3dm1QDPxt4pEaxLRgJv91jotDie4_P-blCo8i5bnoo4,1560
maxtext/configs/post_train/distillation.yml,sha256=GFTNj0w3x_Z8F0UtcSwdn-qNpdmElt7LQ1KJUBCsFpE,1661
maxtext/configs/post_train/dpo.yml,sha256=0-y-RV8S19ASzbuV5vMYNtuQ6uz8E8IOX9TSKXASuhc,772
maxtext/configs/post_train/rl.yml,sha256=gp68hAbMJnJAAE0MpKCBpu9ycrzViwAzKdWz_o5P5Og,8710
maxtext/configs/post_train/rl_mt_jt.yml,sha256=LWiKUDLaNSfgxd0k43TTFMFSjg11K3jyCAyFOtrDgsk,5266
maxtext/configs/post_train/sft-vision-chartqa.yml,sha256=pQndTyXN1-myfLVUWWLyc-w0F0G9RV1thOFt0DwkxBU,1226
maxtext/configs/post_train/sft-vision-slidevqa.yml,sha256=JuYsHhY2ows3pYy1SHo3rbA-j3oSU5FrkxLoDCL5-CQ,1750
maxtext/configs/post_train/sft.yml,sha256=dtgiIyQu6VVYdW2ZtkDs46y5bAuR90mTRblSzT-RIIk,1072
maxtext/configs/quantization/README.md,sha256=VtdOXWMn1iPv-Wbpa4JEEKFHNS3ka0xAxO3hL2J2KVM,1471
maxtext/configs/quantization/dense_llm_subchannel.json,sha256=0xla-5pX3LLFnZxZ6kpYZ8DKnSAGxL6gToWm3qf2V98,250
maxtext/configs/quantization/dense_llm_weight_only_scale.json,sha256=ivPXJ5ZqRA0umVpYDBYepD7OgeIsv0dBhBlaEd0yFuQ,232
maxtext/configs/quantization/int4_weight_only.json,sha256=oKNQduLQF38glRTzYVfH2tpFKOCE6StGJIgP_U3CLJA,34
maxtext/configs/quantization/int8_weight_only.json,sha256=75u1lxq4LHGumkoOSZwckwcOgIUE4H431jAmfVWOFuk,34
maxtext/configs/tpu/tpu_smoke_test.yml,sha256=q20Ea-ytIf0blhjY0AbHSCE7dL3e9RukHh6odt61xZY,273
maxtext/configs/tpu/v4/22b.sh,sha256=4aQTBsM0Ae6FinitHQymt3_3Hr2hh5D4UkMABP5xGrY,2432
maxtext/configs/tpu/v4/52b.sh,sha256=SZg9lKlmNgELQUs39TPfcDnIIh7QdhwnbZySZlR4Jc8,2495
maxtext/configs/tpu/v4/README.md,sha256=MA1qH_XaDpC12OIDbe06SEKs-8EeZGi4HjQyoxV7Aqo,1292
maxtext/configs/tpu/v5e/128b.sh,sha256=Lm_kq6Ey4RWCTIkQPvh3FP0xr-pLRqjdy5La5tr4s-E,2254
maxtext/configs/tpu/v5e/16b.sh,sha256=mhuW0sFCtx00l0Oi-JYUqF-ciK8uOc0-ezstUVRN5FU,2167
maxtext/configs/tpu/v5e/32b.sh,sha256=mGMBFVTA1pDRjTrwe71yuHvwK8_FcGwTXF4231QZnig,2166
maxtext/configs/tpu/v5e/64b.sh,sha256=9jsxxGQkQn4JtnRF3jOhZp7q0tsTsRv4Jz5u0-Y54NI,2166
maxtext/configs/tpu/v5e/README.md,sha256=nVDl97pgUVNIG3Sj-uK4PKRllbXqmx-4fcY2n-dKTqQ,1878
maxtext/configs/tpu/v5e/gpt3_175b.sh,sha256=Fm2pxGQEgKrWFxK59OoELxEhNLM6w9lAb4Wxfh7uGRI,2189
maxtext/configs/tpu/v5e/llama2_13b.sh,sha256=lBigWlub3f37PhwPaZkT1TJX6hyiODNH7QkL-eFIC6U,2184
maxtext/configs/tpu/v5e/llama2_70b.sh,sha256=sbtTLZoSNCY8mzBKDWKgrYJzcJKvxwfzooddrahjywg,2184
maxtext/configs/tpu/v5e/llama2_70b_v5e-16.yml,sha256=kD9XYQcyaFS8awGSR06vRDzyoxJVbbR8-Mmqc1rCa1w,1932
maxtext/configs/tpu/v5e/llama2_7b.sh,sha256=nN6d1rzFHiWDUt7WMVJLbNQ1Cv1v6zjap1uqXQvmjQQ,2174
maxtext/configs/tpu/v5e/llama3_405b_v5e-64.yml,sha256=1s6c4m8Sw3GhYBVHW_lsexEnJ59wIzV6Uz-MHdvmKEE,1805
maxtext/configs/tpu/v5e/llama3_70b_v5e-16.yml,sha256=br7OH5edxaqB2kfGfycI8cXjGI24hnFezMi97Aut92A,1983
maxtext/configs/tpu/v5p/1024b.sh,sha256=pIGs9R8kqzNsw6RkBJpbJZJhUmeFnyfiOqPDgzPeTSM,2390
maxtext/configs/tpu/v5p/128b.sh,sha256=8CpvcZBNk7eVc6K4jQp45p5dYDSVcWN4OdMR3v01ysY,2384
maxtext/configs/tpu/v5p/256b.sh,sha256=f9JHbWUD5UTKJbluCh2y3RJ8YmNwK36FZto3wXzbtQQ,2438
maxtext/configs/tpu/v5p/32b.sh,sha256=YKzVkyixtdpOfRaJUMWfUsfqY2pkGsmZFU3wJ40ksrU,2580
maxtext/configs/tpu/v5p/512b.sh,sha256=Ei0FnPBnlQl1Bc-WDDJKJQk1oHKr656OninBUXQRiag,2435
maxtext/configs/tpu/v5p/64b.sh,sha256=a6W5CfpdmuOnRJNQcg1ESsBBHGjAmktdWV9_XVAWGRo,2368
maxtext/configs/tpu/v5p/README.md,sha256=ydRcAfq3iBZZQ4DDTBOZYTuNYWUB6WQqPEujmRKXlFs,2167
maxtext/configs/tpu/v5p/llama2_70b.sh,sha256=DBx7ao9sgY4ryws8v7u0WoJ2ZWnqrh9u68sxlGoS2KA,2661
maxtext/configs/tpu/v5p/llama2_7b.sh,sha256=ROAmm0Q57uGAm7iVeAJDlHIiFdy7Aq7JWFIRdX0eVIE,2731
maxtext/configs/tpu/v5p/gpt3_175b/gpt3_175b_base.sh,sha256=w8So3cMDFSuvuPl3fBtGzE_3XCFx_KXey3sWjC_4UB4,3925
maxtext/configs/tpu/v5p/gpt3_175b/v5p_1024.sh,sha256=d3WYnZ-Jh0TL31tb2HmAe55G-EUBEkywV1HYSQqS5XE,604
maxtext/configs/tpu/v5p/gpt3_175b/v5p_12288.sh,sha256=-oo439WWI9-McfGif6VDGooRlYISDhhCRgG3tyCJ0EE,629
maxtext/configs/tpu/v5p/gpt3_175b/v5p_2048.sh,sha256=1ltNAUpk79qUprN23X3mmlDdpudcisGUTIMgGkhtamw,622
maxtext/configs/tpu/v5p/gpt3_175b/v5p_3072.sh,sha256=Xoe1igFUQHRf464UIX6RoioIzNi5KMhjX_5vibS-8uM,623
maxtext/configs/tpu/v5p/gpt3_175b/v5p_4096.sh,sha256=3znmoHVMSGACxn7bWvRJTQLCnM0Y0wNi4MNuXgMeIW0,641
maxtext/configs/tpu/v5p/gpt3_175b/v5p_8192.sh,sha256=K1X1jzUfru1PfksxraOR9d9Qra9hIXrwEk-cxDe3cHw,643
maxtext/configs/tpu/v6e/gemma2_27b.sh,sha256=F9XgGziuPqcme7BSrGpu74B_RHwHGQusg5JSC_C6D98,2106
maxtext/configs/tpu/v6e/gemma2_9b.sh,sha256=EUvtQ8XxkI5mY6yMLXVT2TD_NNlUHWZPNu0FlFPJ7Qs,2102
maxtext/configs/tpu/v6e/gemma3_27b.sh,sha256=UsKEuEgOLqdWPZJI3ufkxD3HnmRoj-CM7x_5C_tcBnY,2106
maxtext/configs/tpu/v6e/gpt3_175b.sh,sha256=nFX2Wj5AgoN3970TIOJ7wBXIU08ugEh6xzUQogxgIQg,2119
maxtext/configs/tpu/v6e/llama2_7b_4096.sh,sha256=5C50H8RqbwgQ-pwFpGTLJ4G18zXg53sPCUa01lewCOs,1956
maxtext/configs/tpu/v6e/mixtral_8x7b.sh,sha256=ktyPPa0FaT5lMQxaeza6ovR4FKe8B92uQDnA7_x6vaI,1960
maxtext/configs/tpu/v6e/inference/llama4_maverick_v6e-64.yml,sha256=DaPJcGVWvGDWYCg2lBMKDfyRm1EVvEX4nUSXLeZuKcQ,2757
maxtext/examples/demo_decoding.ipynb,sha256=VxQnCibw2U9l0F9JydrTbFTyNdrxKCeOJ_X2TOYdIwE,13519
maxtext/examples/maxtext_with_gepa.ipynb,sha256=NFqFQP8x95AOT3OkfATNTheqblBxCOqOLCnX3LYDhy0,24374
maxtext/examples/multimodal_gemma3_demo.ipynb,sha256=y0fSCLMma5PGYRxUZvrKmHaAfwMUbKFSii_s5pPBMrY,7214
maxtext/examples/non_spmd.py,sha256=to9A6lW1aOfdJlQna2IIkVQ4FnrfD5FxYLm7JqlSfP0,1832
maxtext/examples/rl_llama3_demo.ipynb,sha256=yFJT08Fh-90AOAZ5alrsO8REMJk4otC4bDC-D23lbaE,10784
maxtext/examples/sft_llama3_demo_gpu.ipynb,sha256=7YS7a6j-4Z_yQODg28x8bK3E4D2iN2tXUUXRAsvWRWI,25416
maxtext/examples/sft_llama3_demo_tpu.ipynb,sha256=Y35R-HvoaqM_B9E6JLBqnK3pFc71PxpmtcQ-CUxnWcw,10903
maxtext/examples/sft_qwen3_demo.ipynb,sha256=YtBnpphRcH3XUDo1BAPhmGqegdu141_vl2HcvbQi8X8,17555
maxtext/examples/sft_qwen3_demo_utils.py,sha256=UjXMkvpcTVRZg2IJy1qz-InOfoSI_3wSGU--A-YWxRw,6994
maxtext/examples/shardings.py,sha256=G6K-MYU9hu8xM3gU6EDry81vJDMFeiIbw7i1AXAz_jY,8901
maxtext/examples/shmap_collective_matmul.py,sha256=2wY-bdfp3DzBypIFkqlhnE1n0DRQDL0afiU68vm_p2k,7081
maxtext/examples/chat_templates/gsm8k_rl.json,sha256=3qynSoTQjVdlfDoVgmWwCKi5N5PZPTmw8eK-4I2Yg2A,394
maxtext/examples/chat_templates/math_qa.json,sha256=AW87IT8PSe5Vf2tIWUz6JQhsO41MkrOEDZJoDVBrMXw,503
maxtext/experimental/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/experimental/agent/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/experimental/agent/ckpt_conversion_agent/README.md,sha256=LVZAsgoLYfEtH2OimdjW50xsnYXh6WMqezZTVZwYJKg,6772
maxtext/experimental/agent/ckpt_conversion_agent/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/experimental/agent/ckpt_conversion_agent/analysis.py,sha256=15ZxwnNC7anlCDP9bZhzaBubnYVJOK4mBPPysjUAh0M,3261
maxtext/experimental/agent/ckpt_conversion_agent/base.py,sha256=ULYelpYh9J0GL6FYbLQhac56PReKI_wqavpOOieKeks,1752
maxtext/experimental/agent/ckpt_conversion_agent/dsl.py,sha256=bw2S7LkO5Vcvt-byLCFmmD31UB2Ih-Xb4ASzHy_f4MY,3378
maxtext/experimental/agent/ckpt_conversion_agent/evaluation.py,sha256=wvREbioJMTGgVE9kKS3SdZWpt2S6qeHZaZVFUlck-1o,3840
maxtext/experimental/agent/ckpt_conversion_agent/mapping.py,sha256=UI0zFxssz8F7sXzGdCzrX151CIjNys1EZvG0pBOhVGo,6828
maxtext/experimental/agent/ckpt_conversion_agent/plan.py,sha256=E1lj_ECwQj36VFm_2lK8BN5J4ebl2pZ4nCRaGOT19TA,4330
maxtext/experimental/agent/ckpt_conversion_agent/prompt_chain.py,sha256=Mo3rAVx5JaRxzRmtcgMTCChkxiPWiCxC49m7rcVm6rU,7207
maxtext/experimental/agent/ckpt_conversion_agent/step1.py,sha256=Vu0Wrklqn9CyX8dQaJGZfQP87JrF5IteVW2W5-3cSNA,1807
maxtext/experimental/agent/ckpt_conversion_agent/step2.py,sha256=j5GaWddHuMYH-0cwurcY00w0on1_D4OKSWUptI0524w,2093
maxtext/experimental/agent/ckpt_conversion_agent/transformation.py,sha256=XjHjaDRdCioh7rVwetkRq5TxTvch34tD_M0kUvU6UiA,3787
maxtext/experimental/agent/ckpt_conversion_agent/baselines/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/experimental/agent/ckpt_conversion_agent/baselines/one-shot-agent.ipynb,sha256=yETae3f6kYqJkau4E0f63yFkr0Tz4wTUu29FLWxRMkk,23124
maxtext/experimental/agent/ckpt_conversion_agent/baselines/context/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/experimental/agent/ckpt_conversion_agent/baselines/context/hf_shape.py,sha256=5eO6dTCNOKSS7Q104-wsEO7HLjkq5HxOrWTC86mX2Uw,2778
maxtext/experimental/agent/ckpt_conversion_agent/baselines/context/param_mapping.py,sha256=6pRlfJpt4cjwlUR6f2J8q6_uz4m0a9Bl-KaV0RCgl9Q,17081
maxtext/experimental/agent/ckpt_conversion_agent/context/dsl.txt,sha256=5dx5bTMcGs1t1gu0hcCHtQjXaZ0pjHFqV84rznfPY04,2453
maxtext/experimental/agent/ckpt_conversion_agent/context/gemma3/hf_params.json,sha256=524qW91an324UJCx_fsUqUm7aby83wXZ4JLmxq7aPn4,77830
maxtext/experimental/agent/ckpt_conversion_agent/context/gemma3/maxtext_params.json,sha256=Ey94lr7uY_E-81qOKMctk0blsvhY0ZRuSeYwCZb-v8o,93892
maxtext/experimental/agent/ckpt_conversion_agent/context/llama3/hf_params.json,sha256=h501PeE4miXYQau-sH0GjLp-hjG50TT-lRhL3lBxSUY,19834
maxtext/experimental/agent/ckpt_conversion_agent/context/llama3/maxtext_params.json,sha256=TI53O3wj6qswOlanLQkgZIDkxvEtVnLmQz9fu0Kc2nw,23780
maxtext/experimental/agent/ckpt_conversion_agent/context/qwen3/hf_params.json,sha256=2QKoxFtGwdHVMa8xruc9SAbNopApIIEtmjUEdcmhsSQ,26498
maxtext/experimental/agent/ckpt_conversion_agent/context/qwen3/maxtext_params.json,sha256=HbuSY_3i52muF4SYfc_n8kcFBt7OjxjRa40MCLaIxmI,32016
maxtext/experimental/agent/ckpt_conversion_agent/ground_truth/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/experimental/agent/ckpt_conversion_agent/ground_truth/gemma3.py,sha256=aAh2_QZwkkzvwz1lENhRu5v218iFbvvqeU_juAXpehI,5916
maxtext/experimental/agent/ckpt_conversion_agent/ground_truth/llama3.py,sha256=2tnu7ieoPo8Yy9wANuMdt6ofMP5wKhQPKRcUd1R7R1Q,4301
maxtext/experimental/agent/ckpt_conversion_agent/ground_truth/qwen3.py,sha256=TKY6yEf_6PtUyAeHCHxm-Lc_FBhuDTH-KfJTZjFkMfk,3554
maxtext/experimental/agent/ckpt_conversion_agent/prompts/01_analysis.txt,sha256=FGhoWZDDcJ5H4tFay-wA7MulOtg1tGuY3XEitaT5ya8,4363
maxtext/experimental/agent/ckpt_conversion_agent/prompts/01_plan.txt,sha256=SDryfH2mlNN6OMlHJGRBI-i-4u2yjpETbymXirl1740,940
maxtext/experimental/agent/ckpt_conversion_agent/prompts/01_plan_check.txt,sha256=HayFSgYce4kxCLSIJPTDrQudYHP2V_1_AOKSDCexg-A,374
maxtext/experimental/agent/ckpt_conversion_agent/prompts/02_transformer_config.txt,sha256=MRSJd1RVfrI_1Z6jbBjEtw3yqapUN59jC8M0FmzYhFc,1093
maxtext/experimental/agent/ckpt_conversion_agent/prompts/03_param_mapping.txt,sha256=V2EOIqUveWo1zvQb6X9AE7IWSxY-B8srqIT72LRKuQs,2852
maxtext/experimental/agent/ckpt_conversion_agent/prompts/03_param_mapping_check.txt,sha256=3887iQhDg8rgEXO2qtCLDetGipRK1QV4DR6KwizdasM,1384
maxtext/experimental/agent/ckpt_conversion_agent/prompts/04_dsl.txt,sha256=FvqlWtzmNn3Tp5yBdNmZ_1MmpxrWz259tojUay-Fuo4,1535
maxtext/experimental/agent/ckpt_conversion_agent/prompts/04_hook_fn_dsl.txt,sha256=aZ7ahDKLxFZXrzp9BSn3V7iFIOVuUaDHp5FKCdkQB4Y,2864
maxtext/experimental/agent/ckpt_conversion_agent/prompts/04_hook_fn_prompt_chain.txt,sha256=6lu02IDlVKO6cVTzLFrpGHn9o8Kn2vYRpxaHt8xaj3A,5161
maxtext/experimental/agent/ckpt_conversion_agent/prompts/04_pitfalls.txt,sha256=njesPrUpdK-Q7OhJx_uZqFSnCh_iqDSCBDg4jVBAB7g,792
maxtext/experimental/agent/ckpt_conversion_agent/prompts/05_shape_mapping.txt,sha256=NO3D9FNyE7IGdhG6ECqqUragqFhcdgaKZFJ0mlVkmkI,2064
maxtext/experimental/agent/ckpt_conversion_agent/prompts/05_shape_mapping_check.txt,sha256=dCQ1xt82dmnoOEHQ_-Kd3i0Pe6-CD0bNzilmQhwSWVc,198
maxtext/experimental/agent/ckpt_conversion_agent/prompts/rate_outputs.txt,sha256=a0ZwwjItQhOsz_OphQK3m0xrn7v5TxwJh0pvcWcNwaE,2585
maxtext/experimental/agent/ckpt_conversion_agent/utils/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/experimental/agent/ckpt_conversion_agent/utils/save_param.py,sha256=VymeEyJIEWXNQInz2XrooByqaDuBWqn3hVw5AlxttGs,4781
maxtext/experimental/agent/ckpt_conversion_agent/utils/utils.py,sha256=5cCUPRJm8YgkiSPZ1y1SkBiK4Gzxf0ZvKSsm5z7e8bQ,2129
maxtext/experimental/rl/README.md,sha256=gYKEIqGx97U8f9IQdfqKm3MOSpjyC8lk24Y3qoZ8Z4A,3869
maxtext/experimental/rl/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/experimental/rl/grpo.yml,sha256=0d1rX8wRf1kmUu2qoSibjN5Gdb79fLoNvMrIuvJprR0,1034
maxtext/experimental/rl/grpo_inference.yml,sha256=Jv4Qn-f-VUHPGYO2u1Uzfq2nWTFFrAAtOPNmIUmP430,1022
maxtext/experimental/rl/grpo_input_pipeline.py,sha256=_4PTckhBh_vghvdKfFIR-X-bvfoyiAGk1XnGw7iZI9c,9368
maxtext/experimental/rl/grpo_trainer.py,sha256=dCAUdlqROU9lf2VrGlBbOEPI-nJ-BUvo8krNbr11Qzo,41302
maxtext/experimental/rl/grpo_trainer_test.yml,sha256=zC3UqWsJvs795st6zTKaKZDQsfxkiU0g6pVK97dMQso,318
maxtext/experimental/rl/grpo_utils.py,sha256=T-1rYRHZ7HsYBhoL0B4Wy06T_1F3-1d7qCgmlJ2_WUg,23290
maxtext/inference/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/inference/decode.py,sha256=J4G72PzGw350QSdMalgr7DSGhwt0lc_SLidZ39nJsBM,10128
maxtext/inference/decode_multi.py,sha256=nBsNKNyWp17OxksBxRW1dqffxj4KxwGTrr_CrQybOzM,8814
maxtext/inference/inference_microbenchmark.py,sha256=ClYvZUoEboB0DkE0WhL0RdXmHOm17zQu3osNYPTXkHc,17943
maxtext/inference/inference_microbenchmark_sweep.py,sha256=HWHYC_tkUiw5BWPDxNJ-hLONQtcXw1onv7kXh6Sa3G4,6415
maxtext/inference/inference_utils.py,sha256=ps5C4YcbAI1cDoy1Qcprf0hOrDGDqOhrVzhZEyLSd68,9874
maxtext/inference/kvcache.py,sha256=u6DU6S1WpXSVF7JdWUtX5CaxfzbagviaZxiUgEm0J2U,44144
maxtext/inference/offline_engine.py,sha256=3hO2J3WvaYPPAxwdUZEu_EXjY7uXNHbsSIspJ3j8MxA,32355
maxtext/inference/page_manager.py,sha256=DnNiiwcIqbQCsEgy2JRx_ZCisYAkwQM_7_3zJ6Xz82w,25522
maxtext/inference/paged_attention.py,sha256=lL3YJKYKy1uIVwLfH9aSv54tREnVbTEQWTJp6ovHinw,20379
maxtext/inference/paged_attention_kernel_v2.py,sha256=TuACJllMLxTiQE08idI86zBA93J_sxTdEbZ3klyeY9g,25380
maxtext/inference/vllm_decode.py,sha256=4e4mXIL7bmaOR9qQj8tLkHdybxnCB2LBLxmELEKxpQs,9210
maxtext/inference/gpu/README.md,sha256=JnQyfxEyZ2l8aX4PME6O8gXb2Fd1dJq66kbZFJqvASU,110
maxtext/inference/gpu/microbenchmark_llama2-70b_h100-8.sh,sha256=Q9irE8l4y5BiNz2UpbISYa88RNsKurobCN4NW8BqN54,4337
maxtext/inference/jetstream_pathways/README.md,sha256=RSsAQBSlYnfizP_hn1bGXX0LsT-iC-kau2x7c6QUHX4,829
maxtext/inference/jetstream_pathways/jetstream_pathways_entrypoint.sh,sha256=DuO5L2jSLNspkI2Fvyt5ER8StgfqTkwNNVAoT9g2tJY,666
maxtext/inference/maxengine/__init__.py,sha256=FLQndEGUQP5UJqN9l6no0Qk-menJBMPd73a8W0IwES4,616
maxtext/inference/maxengine/maxengine.py,sha256=nrSPNMonpM5h3KjU5RsbLai6DyXWwBdrkw_yrbVn9WQ,72648
maxtext/inference/maxengine/maxengine_config.py,sha256=d6BuAiQgJa-7GVHu6AeM9-_l51fnTpBFDrnr5IRy4h0,2936
maxtext/inference/maxengine/maxengine_server.py,sha256=UHmLWGnNaT8Q017E78ukdJ857txrP80D0WRh1PnKWXI,3942
maxtext/inference/maxengine/maxengine_server_deployment/README.md,sha256=rAM9S4s_sxUiY_252i1zJCqaOTb2TiFajvvt0tuOObg,779
maxtext/inference/maxengine/maxengine_server_deployment/maxengine_server_entrypoint.sh,sha256=cQOyVCnnQhDWeOY623s8G3Aun9NldJN9Hz5vdsKoePE,693
maxtext/inference/mlperf/README.md,sha256=CJaZMSYLm7GJ-GB_2ChkHP6VxEheHT3FdYjGmwmYQH0,8100
maxtext/inference/mlperf/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/inference/mlperf/evaluate-accuracy-fast.py,sha256=yShOvyOoeDxZaLoqT7UZ6S_J-VPCjn0uquhutni0_8I,7011
maxtext/inference/mlperf/evaluate-accuracy.py,sha256=YEeZjlHUxylKjt9dhSKRoD5BW8XU7rqO7M-qFBHxZrA,3912
maxtext/inference/mlperf/llama_offline_run.sh,sha256=DtEvH-lSqhzDZexrYbVg2F9Vtl5EvLEvw1nDYi9a5k0,6212
maxtext/inference/mlperf/mixtral_offline_run.sh,sha256=0xSF9bhPS6EwQCEyCIipF_6jdAHEU1qCboDS7BcFYJo,5913
maxtext/inference/mlperf/offline_inference.py,sha256=QFuHV6oLGBKYCMa1kq_07r1zCgqnVrQB-2ApN6C7N4I,13592
maxtext/inference/mlperf/offline_mode.py,sha256=8dKFNlcX-wkdpxDMe0Hx_pxJLVOAQhmJZH0L-VAqsUc,17622
maxtext/inference/mlperf/requirements.txt,sha256=x-FmEA13pP3Hdd9pRffJrdYFyUc5-mX7kKadVo6JPhc,134
maxtext/inference/mlperf/user.conf,sha256=tzhHLB7VQU4XrbkiRq_ZqzWLZIyaps8WdHcPEpdhFm8,864
maxtext/inference/mlperf/user100.conf,sha256=lJn2hnEBqpIIz5cf603ltLHq7GoTXfNnidZb0MxV-v4,986
maxtext/inference/mlperf/user5000.conf,sha256=9nnodJ35UH2hB9YDhxsbURv34zWpQfBS1A145ohwRCo,991
maxtext/inference/mlperf/gpu/benchmarks_llama2-70b-h100_8.sh,sha256=Utj0QBcxxtALW1GW26S750hOExg6p0mb2hHIzdcSI1s,4062
maxtext/inference/mlperf/matmul/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/inference/mlperf/matmul/matmul_dtypes.py,sha256=Y2fHq4LHzrmFB4HfZbVHrcl_yX5a6ifQNzYKR7LT4jU,1527
maxtext/inference/mlperf/matmul/matmul_sharding.py,sha256=fWsJemmgIDcx5zir4pVo9_KPI_HIHoDJchKrBk_xTtQ,2586
maxtext/inference/mlperf/matmul/timing_util.py,sha256=_jn8Seys2q4FWlPlttGeImbq5r6bNkWSYO4BO0L31yc,1583
maxtext/inference/mlperf/microbenchmarks/__init__.py,sha256=6zhOmlklLlWXvz95ApRtGXDV53TEZcoEA6V7y1iWIEc,619
maxtext/inference/mlperf/microbenchmarks/benchmark_chunked_prefill.py,sha256=9iaDDeBi4S3g-CXTpocYOOAg6b-F3H7pzO_DynfJIVM,14907
maxtext/inference/mlperf/trillium/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/inference/mlperf/trillium/benchmarks_llama2-70b-trillium_2x4.sh,sha256=ydraoRh2du5OUTw0n1ula-n3olswrqeUrdwV8-OyJ6k,4000
maxtext/inference/mlperf/trillium/microbenchmarks_llama2-70b-trillium_2x4.sh,sha256=oLJdELjM65CH_Kyi0PDTRJRBAnRSsfTdQ3Rig2bxLCY,2766
maxtext/inference/mlperf/trillium/select_xla_flags.py,sha256=rJwFsGi3s7-00jWh7Yljxuea_bJZdkSL0cjFAQH4duI,3581
maxtext/inference/scripts/decode_multi.py,sha256=nBsNKNyWp17OxksBxRW1dqffxj4KxwGTrr_CrQybOzM,8814
maxtext/inference/scripts/sharding_utils.py,sha256=77pXW-hcDgO4f-iDL6hF2MjbTsbTWHgPgb68q91SDhI,17488
maxtext/inference/scripts/test_sharding_utils.py,sha256=rQpMVoSvhz7BVMCAs_3AEEYLTgRlz86PMiM4uVEYV8k,17840
maxtext/inference/scripts/notebooks/sharding_utils.ipynb,sha256=I-Bewe6RdY1JXePhDXvVWRQJRZ2zFjaDbGduPbJtEAI,529210
maxtext/input_pipeline/__init__.py,sha256=otX-EDUVsL8kHS275NFH8VnNE-7RDyQmKir2uN1UyGg,581
maxtext/input_pipeline/data_processing_utils.py,sha256=gX9duSANa7a35ilFrc-pFTAJollXimpLeGhFY64LOdE,6710
maxtext/input_pipeline/distillation_data_processing.py,sha256=m5g2Y7qtAzQmte7zEIFNA8-1qR5iSiRQ4pPfQ98-_lc,5419
maxtext/input_pipeline/grain_data_processing.py,sha256=RQYKTTqYQjqmaEqf-RkXfAxisI1TKkthHdr5nKmjme8,21183
maxtext/input_pipeline/grain_tokenizer.py,sha256=pAyUIQKFhSFrR-tWtxUW935QjBJMC6skk8drLOGBN7k,4074
maxtext/input_pipeline/hf_data_processing.py,sha256=NXlFPu9ViOQ9SNP-krUz104u9Ae4WQBZ-qEJ7r43De4,18133
maxtext/input_pipeline/input_pipeline_interface.py,sha256=mMUXnj4ZYmZVUiDN5UIGVm8OgNheMo6OVO0WmR7cfqw,5826
maxtext/input_pipeline/input_pipeline_utils.py,sha256=QvOvw1_hi_hYrWype56_PSujpxw1LNnKnh7f47g4PYE,38711
maxtext/input_pipeline/instruction_data_processing.py,sha256=bc8lgdOFeceAZKk74DkSxDBRfS96PHXrVQJN0tx_WGs,7550
maxtext/input_pipeline/multihost_dataloading.py,sha256=wMXZUVUfwnAH0P7zRqAWmNX-KEaJQTvg_JpIRzojnsI,11531
maxtext/input_pipeline/olmo_data.py,sha256=pOJDZqwITZTiH0ANJXk2IhwpGBM87zFKLdJnVxgAr3E,13879
maxtext/input_pipeline/olmo_data_grain.py,sha256=nQYvaQxqP5jyKAOAVDUJSvXt3sJvSLEC_c0BBXgtmp0,20182
maxtext/input_pipeline/olmo_grain_data_processing.py,sha256=vsUqWV-5IiCE-BfrxoJcToZiPuWPlc3chhmxbRgSGpM,7504
maxtext/input_pipeline/synthetic_data_processing.py,sha256=gdQVbotMqlJr7IqD85b-aFUNmbxcJi0B8z0QzDAkIRM,4034
maxtext/input_pipeline/tfds_data_processing.py,sha256=nfQQn5OyP8qPvq_52RdYhL-Mk4lOivBtn_cVbu8Mvhg,11205
maxtext/input_pipeline/tfds_data_processing_c4_mlperf.py,sha256=_WQX0tipxRLDcP7SNRu0S8NtuiQcLQR9_bZlaCDD1eU,15073
maxtext/input_pipeline/tokenizer.py,sha256=YymD7P5URwRTol44zzb2z03SLu6LJX0yETFNF6F6e6s,8932
maxtext/input_pipeline/packing/__init__.py,sha256=otX-EDUVsL8kHS275NFH8VnNE-7RDyQmKir2uN1UyGg,581
maxtext/input_pipeline/packing/prefill_packing.py,sha256=Ka9stYSYqSPLaQStqMMdeE0GewBdzIPEJcTdo5IfLTM,14505
maxtext/input_pipeline/packing/sequence_packing.py,sha256=wntZlKmz2Cjsx1ytAxcf_sKJDPusuR-YTjnlHjeQ4p4,7880
maxtext/input_pipeline/protos/__init__.py,sha256=otX-EDUVsL8kHS275NFH8VnNE-7RDyQmKir2uN1UyGg,581
maxtext/input_pipeline/protos/example.proto,sha256=gXSdgDygbQ4Z7Zj95E7bpbi2xzFed8p4QU8JyFWvAgU,10752
maxtext/input_pipeline/protos/example_pb2.py,sha256=G95hlluxP1-Jg47NmtP0U90WruxiTC9dhnKZiDL4ZjI,2296
maxtext/input_pipeline/protos/feature.proto,sha256=ax7X62uxA8TSJqoLssNq3H8y-c2zMG-S-4bzlSMV7r8,2769
maxtext/input_pipeline/protos/feature_pb2.py,sha256=hdK6rO0TYHaMtAqOctX7OKOEFAtacOtI11EoJE7dN34,4357
maxtext/input_pipeline/protos/generate_proto.sh,sha256=iamNzqopWsKectXJ1aNq8Gooany6nq4LiG5V373G45c,1448
maxtext/integration/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/integration/tunix/BUILD,sha256=kVEFlhryrW9sm0rFY6bsct6prtmMKBDyJmMr3FNw7Z8,1712
maxtext/integration/tunix/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/integration/tunix/tunix_adapter.py,sha256=obPyU8_ASSTgcd28Hoqvft-ntTfeuGcwY_6ik3pHq6g,3151
maxtext/integration/tunix/utils.py,sha256=FwvOwL9CS6SMPd2I6MzsJqFJm4Si_-o3Kc4Sk8gf7tE,9656
maxtext/integration/tunix/weight_mapping/__init__.py,sha256=AcaBXR6FmSrp2oe4ZJ9AAcQdFwy_tp77TlQbegKiBek,1987
maxtext/integration/tunix/weight_mapping/deepseek3.py,sha256=3VgCC68weuC-lj1PlksNnuvAtZU6nHgKc91mf9ntbe0,6869
maxtext/integration/tunix/weight_mapping/gpt_oss.py,sha256=-_csU5BW90OY44BYO0giywiV-wq_zfv9GBeC6LE0vf4,6044
maxtext/integration/tunix/weight_mapping/llama3.py,sha256=S_bB6MVf5nckIto0IO_fgPzU4ThuARQFps6jGjhNNoE,5918
maxtext/integration/tunix/weight_mapping/qwen2.py,sha256=sv1pm-PrtBPNHygoDwvWofQMP19Pjo5PW2_rkFb8p3w,4821
maxtext/integration/tunix/weight_mapping/qwen3.py,sha256=vr6h0HUThNOMf_yvqWCqFz6Bz9kV-VP8psnwGzfcRkg,4618
maxtext/integration/vllm/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/integration/vllm/maxtext_vllm_rollout.py,sha256=OxneOsPoFlIthnqb5RSPLwqgQdDubAB-8EKOFAYg4oY,8317
maxtext/integration/vllm/setup.py,sha256=elb-FqVRJGL9QXT_vqrk2i_FSyYk-MSYXGmTa1mXf9I,910
maxtext/integration/vllm/maxtext_vllm_adapter/__init__.py,sha256=w2mRQY0jJG7T06boHfg__dzDryHreMOUrGqJF8HhYTw,1243
maxtext/integration/vllm/maxtext_vllm_adapter/adapter.py,sha256=5eKl_hJ3QB7iU8eW_8oOWdB_rGPRy0nGzIATVt-uyH4,12189
maxtext/integration/vllm/torchax_converter/__init__.py,sha256=otX-EDUVsL8kHS275NFH8VnNE-7RDyQmKir2uN1UyGg,581
maxtext/integration/vllm/torchax_converter/base.py,sha256=Nm9RJH0e4U--9WElrN3TSHv7QmgFMbY7K5jIrnSg7H0,2005
maxtext/integration/vllm/torchax_converter/gemma4_moe.py,sha256=WZYwqWMCWc0aQfeDtfTJCNZ3OPPp9j5B_Dv4f34SDoI,17674
maxtext/integration/vllm/torchax_converter/qwen3_moe.py,sha256=jaaUJxRnsgn8u5vtPadaF_KrbqurmlchXCd2iZGmkAQ,9064
maxtext/integration/vllm/torchax_converter/validate_converter.py,sha256=WO6LBRW10btB9OK7JKdwZzcpEMqmSmA9hYkfHcbUpT8,16473
maxtext/kernels/__init__.py,sha256=otX-EDUVsL8kHS275NFH8VnNE-7RDyQmKir2uN1UyGg,581
maxtext/kernels/gather_reduce_pallas.py,sha256=AfCI_LsV9sKAaxlVoOfECAB5pxkOdeufyQIE0dn8kFw,7740
maxtext/kernels/gather_reduce_sc.py,sha256=nCWPdW_Y8VrkzxWVhIrRD2mOilJJkrbEnyS0H7OeX40,56161
maxtext/kernels/sort_activations.py,sha256=UGKM13HhlA_E4vlZPqrORcXIJ0TA1W7L2wX8XUgCsu0,3826
maxtext/kernels/attention/__init__.py,sha256=bMZ8bGzaJ2Vm9NvKkrpY1LDqsy8Rx0Ka8cS2YP7zPnY,670
maxtext/kernels/attention/jax_flash_attention.py,sha256=j8mNM-WGOH4SBCy3X-x6hZuuboxZ__hPlbyEvblX1lM,10326
maxtext/kernels/attention/ragged_attention.py,sha256=Mj3F2zvHVAYd6QlsgwWvAW3PmeS0_1MIhCUhxHAZ_lA,13300
maxtext/kernels/attention/splash_attention_kernel.py,sha256=DUW2QJQXgHO5AFaQeTZSDhxSDl2qEp6kUBFUBNL2ijA,78695
maxtext/kernels/megablox/__init__.py,sha256=nxu9b3_E1SxAK4kUpf_rFUloFeNcDGXJ8CYlrGbD75Y,649
maxtext/kernels/megablox/backend.py,sha256=L-t2J-xmwmNtaexwQgv3xeA-L49aNf-vreFx4QVCrlo,28866
maxtext/kernels/megablox/common.py,sha256=7ofB9dWKhxInTNCZMIIb6CAKU2t1A1GAUGMytuunruE,2044
maxtext/kernels/megablox/ops.py,sha256=0ks-lRZbIMcuX6jR3z2VMfpf3vN0j4TOaaegZcbku4k,11385
maxtext/layers/__init__.py,sha256=otX-EDUVsL8kHS275NFH8VnNE-7RDyQmKir2uN1UyGg,581
maxtext/layers/attention_mla.py,sha256=wv6w-QyTsWFTi0DI4-AzAyhiQPwmjpPcZF0TXqVH3L0,51607
maxtext/layers/attention_op.py,sha256=HqgWAlRhPDWUFLCi3oK2QucXVdI67XDV5DeG6OPc6_8,88896
maxtext/layers/attentions.py,sha256=KrjL084YEKcYDD-xyT_2PkXOjAwMj0k0xOc2rPhu4EQ,50458
maxtext/layers/decoders.py,sha256=XjtHVJ5T0HhgGnAJTrRcWK-U5UgUNQ73wUQFv7hFAkA,56514
maxtext/layers/embeddings.py,sha256=4TurThlgQNwidyiW8CLGuL3nZSKoR-G6uKTxtFE6fHk,68250
maxtext/layers/encoders.py,sha256=K9b5y3UCP67y3yseXzrPRhTecQly53Ej8aF57iw9jWY,6585
maxtext/layers/engram.py,sha256=Xs80DzXPmhpD_AQi9kRkfWh64Y1N6AjHS9ToS-KZEXE,25344
maxtext/layers/initializers.py,sha256=4g7r-iqnMFP3h0lL5Pm_l8HCmL7sz8Wz7Gz_18ODxpc,4023
maxtext/layers/learn_to_init_layer.py,sha256=5PTLLSkPpDaSLhdxjHjO_dgyUu3mz4j6dcYDyIK0eEk,17654
maxtext/layers/linears.py,sha256=7tXGjCMUonoWhLbDpzA0CtmCIwULmQIR0YjVhPw8JAc,20873
maxtext/layers/mhc.py,sha256=4lbNocJpapelsV2L-y3KO-ahLejWXWZIuzsS3u8sdVU,9185
maxtext/layers/moe.py,sha256=uyaAN87N6cIgR1dSo8I3ZVKFxKWFnmyxVMb2zdmQK0w,102103
maxtext/layers/multi_token_prediction.py,sha256=mveCnnYIS8LcASwsNqNyDMskolNUkEJ-oSCn7ypHlME,13424
maxtext/layers/nnx_decoders.py,sha256=G4vX4EY31NkqPEU5c33M8OxhwFafDaKmW_nxjB1x_-g,54863
maxtext/layers/nnx_wrappers.py,sha256=VxvgocNeT4PBMW3xlih81bzSzoe_xZ96TxkiQR-YJCQ,22075
maxtext/layers/normalizations.py,sha256=SAuQoqv5zsLOPUZwYuplHtAZCxbIFogNU63zvlvnOK0,8003
maxtext/layers/pipeline.py,sha256=3GgIMUWuA362YBEqBdVSC3MIpBF7Fye-wWAZ7ot00Rk,63861
maxtext/layers/pipeline_deprecated.py,sha256=S2Eyyl6lmvqW5X7-lK6c-5vOlywAuRNIITetYsGjY9A,44245
maxtext/layers/quantizations.py,sha256=iNAgKlwMeyVaPhDV3xTOatLLEavXMp-4Ff6WPiWjfbQ,37655
maxtext/layers/train_state_nnx.py,sha256=t6xuqq_lJ7WRtyLM67qaCFV_UaWl3GJs5vukMxGj3Es,1612
maxtext/models/__init__.py,sha256=otX-EDUVsL8kHS275NFH8VnNE-7RDyQmKir2uN1UyGg,581
maxtext/models/deepseek.py,sha256=34sOJW1MmbXi0UiFVWeveUeEaeHjrmO8wwQqxBIL944,22707
maxtext/models/deepseek_batchsplit.py,sha256=XM0cHT4_O1FXhPrAX9f6bysCpdPuBnrlrP1F2tUBYJ8,87526
maxtext/models/deepseek_batchsplit_fp8.py,sha256=Us32342rPjObek_xYpaFdAYmPwITTjDqz7HkjVLmzOA,37352
maxtext/models/gemma.py,sha256=xNAHmecpH6asWxtQb4nk6TRNNomasXNGnr5YR5JHIJ4,6219
maxtext/models/gemma2.py,sha256=abImXfD1fgcXfA_e8uLf4DEQgCcyMSuLznS1Ml9msDw,10886
maxtext/models/gemma3.py,sha256=nQGkjbDIJX2rtWQfmtNzcOX58hDs1D70bZoBmBzc3v0,23478
maxtext/models/gemma4.py,sha256=Pti2wEZ9PgPyyzu3rbabzEbKeMPPGlGJfUuHbI5ishg,16407
maxtext/models/gemma4_vision.py,sha256=G0Aw_RPwGteZ8_NN0srN6mR_9HTUuh0_8iQMfZ4gz1s,22172
maxtext/models/gpt3.py,sha256=WiRmYCivJEE47IDFjBpOOxQTGsbIbAuaNdpFAudYBeA,17059
maxtext/models/gpt_oss.py,sha256=DRgDFlsN52YJs4hZHecFboIOEiM-8zi4UawE33pd_3Q,10082
maxtext/models/llama2.py,sha256=nF99u3YI_ezBRdmGd_IBD5dp2DiIrcXoxPGnG-4Ydmw,8888
maxtext/models/llama4.py,sha256=Jx18CuZZpQ9_QNCBsmI6by60lmcB6R36kmSKr8pHVoQ,30652
maxtext/models/mistral.py,sha256=dEG0uakXzi_v7nPYEnVugRDqx69iIuppFWi5kuiLhto,7278
maxtext/models/mixtral.py,sha256=0Uxwun5IyJKym_K4mgYR7z949lhZw9D9VHxq6KEN2EI,7091
maxtext/models/models.py,sha256=ntrDbFhhePqLFwTbm9o2NsAgrXlGcc7V55-rsc100Os,24618
maxtext/models/olmo3.py,sha256=ExDHgfjiYp_vNxuLXJ1zMIyxg5ClxactO8jiJ6incos,10372
maxtext/models/qwen2.py,sha256=yicE3gISAe_9ALTOs7rYCPAAGnUKmOR9k6ER2LfPoHE,7619
maxtext/models/qwen3.py,sha256=HWi6iCcUCyjtYPOLL0_OWNmH84VDEMhM5u1Cy1r4TaM,80251
maxtext/models/qwen3_5.py,sha256=qOSrat2DOXwS3ZbZatyzNxBtnREWRXDZ0TS_vmLkHO4,8424
maxtext/models/qwen3_custom.py,sha256=k0jqzVJzIeaq7uEBbeorOrLzG6uu_wbgted0hW2Toa0,9806
maxtext/models/simple_layer.py,sha256=Oi-zw4fiXxuJYzugGOMOIUa5-mWEgGPjuyfGN1uZ9Tw,4905
maxtext/multimodal/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/multimodal/processor.py,sha256=iX2jZvEn9aweN8yNcErqS64m9lYXyINX2914MhXs9F8,11741
maxtext/multimodal/processor_gemma3.py,sha256=X1zk2Sm1TTHXn62pX_SvrVnofvabiAxWA3PqPT5bBW0,9771
maxtext/multimodal/processor_gemma4.py,sha256=Etl0Miu2c4HxgYIkYT-ug6SnwKlASMxPa_tCVElyR70,6502
maxtext/multimodal/processor_llama4.py,sha256=WjbDO88gq7s07CEpROld9of0J7U1Me2sO3KKgiJN4cE,20577
maxtext/multimodal/processor_qwen3_omni.py,sha256=CTCN67r2BwYuEAuIoWq0h2erNQ9z6Ti5qGRaij5f22o,44707
maxtext/multimodal/utils.py,sha256=C7pV9MhLoC8qtS0687NU037ke1G5nnTDh0pV6XoMIaI,34079
maxtext/optimizers/optimizers.py,sha256=xUiBkA3mMQ-1xj740L1suRSdn7OFcDlpKGGykC2H2vQ,13782
maxtext/scratch_code/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/scratch_code/analyze_sharegpt.py,sha256=iwnyoCtdl89ODuhynV7153UmicCAGpyDOCFrCIRMnM4,4863
maxtext/scratch_code/demo_from_config.ipynb,sha256=qAcdc0xyCrGf8VLFM2W60JETlSKtrN_-Ed2qlec6KN0,38652
maxtext/scratch_code/gemma_7b.sh,sha256=KLxMJrZESR2naSRl8BYYoX589tVTCCJjbBcUusJU-NY,1099
maxtext/scratch_code/mixtral-numerical-verification.ipynb,sha256=uejQ9rh47x5ake91uRvnLq8W5gqd3QufkROCnDy4jRc,11715
maxtext/scratch_code/run_inference_microbenchmark.sh,sha256=gJi4hEF1SSA1DFGRt0TcwMj3V3LkzRP9SxgiqZjPGmI,645
maxtext/scratch_code/setup_transformer.sh,sha256=i8IOZMPzh26VbufobKM2kd5-85l-GJu09mnOqUanTE8,189
maxtext/trainers/diloco/__init__.py,sha256=QF_BPHJWccoZeSaarVzn0MStInlGsQojtUuScrvHPcE,579
maxtext/trainers/diloco/diloco.py,sha256=1S2S8xwQDqbya1qgMobbJDSATdkgvmdQGdTT4LrY4Lk,11122
maxtext/trainers/post_train/distillation/README.md,sha256=E5EGrrBLMkHL4oaBGeYDLdWmexT9mS0sviZjpdtdXSA,5984
maxtext/trainers/post_train/distillation/__init__.py,sha256=QF_BPHJWccoZeSaarVzn0MStInlGsQojtUuScrvHPcE,579
maxtext/trainers/post_train/distillation/distillation_utils.py,sha256=GjYC95-opkIruSu-9ud0HZY8FW8zEVMZ4vKABMKewmY,33613
maxtext/trainers/post_train/distillation/lti_utils.py,sha256=8Bhbr8Xc-pxJkjLhhuzX94311E4efRzcaP7Q3u7S5Ds,4804
maxtext/trainers/post_train/distillation/save_top_k_teacher_logits.py,sha256=GB653AbHW5jfTNNVlOj6Iy5g0jYY4wqdZWQ4qGPk6oc,8654
maxtext/trainers/post_train/distillation/train_distill.py,sha256=z25ZbO3zlx8tc-mrYQxlFdqY8jJ3lJth7TsYNH3URR8,37945
maxtext/trainers/post_train/distillation/verify_saved_logits.py,sha256=8JTqh6s2QEwVvIaqjIpnTmn2Ks_oGy7uEAroVRNhbK0,4520
maxtext/trainers/post_train/distillation/scripts/run_distill_xpk.sh,sha256=OBSMNBJ6b9wfFU_C8H1xbrhDkxWjhHAP8MRMz3LmsjI,19103
maxtext/trainers/post_train/dpo/dpo_utils.py,sha256=V180NliKv1-absGtJS_HpxbPQ3ZR1THgWH9ggHnML70,6968
maxtext/trainers/post_train/rl/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/trainers/post_train/rl/evaluate_rl.py,sha256=omnvkb61NPZ3SneQP_oA2dRIXE3G8AodgplZRfM7wXQ,9527
maxtext/trainers/post_train/rl/math_verify_pool.py,sha256=6LYF8ZnJyvb_gIWhYaVgr3GGD4bc0Grcm4zSmXAc5JQ,10533
maxtext/trainers/post_train/rl/train_rl.py,sha256=cvO0ceV2Q9plrFVPLXrZ8F7D4rwE-wtcQsniJPPWvZY,25709
maxtext/trainers/post_train/rl/utils_rl.py,sha256=XGxB5mwCAMkRVwTO9TWswY1x2UmzHryRvG-zyyuUv54,25464
maxtext/trainers/post_train/sft/__init__.py,sha256=QF_BPHJWccoZeSaarVzn0MStInlGsQojtUuScrvHPcE,579
maxtext/trainers/post_train/sft/hooks.py,sha256=uH1pnqH_W2hGisf3oXl8WG4FgOoDW9Lv48kls0XFlbU,8154
maxtext/trainers/post_train/sft/train_sft.py,sha256=9balf_c9kVegG6k9p77DmOHm_HOBcwrzQaNL26WqDzY,7800
maxtext/trainers/post_train/sft/train_sft_deprecated.py,sha256=1jdqSV2NoMHL4R0U5jnf1JJAO7-Hu7q-LlUke5RxG6Y,7261
maxtext/trainers/pre_train/__init__.py,sha256=QF_BPHJWccoZeSaarVzn0MStInlGsQojtUuScrvHPcE,579
maxtext/trainers/pre_train/train.py,sha256=aA4R5MAzhluv_C86VVqAggqkhLX_P6rJUr_w0UOnksI,30363
maxtext/trainers/pre_train/train_compile.py,sha256=rs66gpOyQQnk8sml8ChWnHlJwluBMGjpuH37vAliQG0,13561
maxtext/trainers/tokenizer/train_tokenizer.py,sha256=NBaytLAmoENjpQkMfEvY55HydefWBR9mmp0N53EqHT0,8512
maxtext/utils/accelerator_to_spec_map.py,sha256=fzSoEw6Va37REvXFBGcRfUhstih3z8hzcJi7hIre0w8,26877
maxtext/utils/elastic_utils.py,sha256=mFvwXpdiEwTVnPATlTlD6b8Xez94Jm3DfqSWKK4PTTg,6729
maxtext/utils/estimator.py,sha256=JRgZWMd7fI0CeRgt6iR5LlA6TE2BGMeP87XSvdlNFQQ,15989
maxtext/utils/exceptions.py,sha256=9oooTMFEvfHMTRGKY2eHXeVMWwgP8YL40cWaep41y_g,765
maxtext/utils/gcs_utils.py,sha256=OxOSz0XKfB-PZ6PfcFSRgtxKN0pV7S7HC51ho4LxDww,10836
maxtext/utils/generate_param_only_checkpoint.py,sha256=2kQ6gDPNayMYcfrTFj7Mz2iq7hk-kidUpBZ4qQaF7qc,9349
maxtext/utils/get_flops.py,sha256=-vzr6Tac8V9udGuVOKtuF3qTISgM8LRM0jrux2683ow,1368
maxtext/utils/globals.py,sha256=PYra6m-cl_OMIFUL960PnU2TWQGuDWQZzI-tcJxYw0k,4208
maxtext/utils/gradient_accumulation.py,sha256=fHF1So8vwbpqeD2r8Cmf7X7z8mT0nGt2rTzM654TlrQ,6438
maxtext/utils/layerwise_quantization.py,sha256=WMsSYJI_XvxYa5ilyKWlBQ47n_PS0au-XJBlLGXwOAQ,13222
maxtext/utils/lora_utils.py,sha256=mBc8DRjj6LdZz9523693N-N9QfwVDJSgvrhOmt-nQq4,14239
maxtext/utils/max_logging.py,sha256=xhab1Cm0xA-YRfnJEacWwNePEv-XxTXoC4nk9TyTqG0,1795
maxtext/utils/max_utils.py,sha256=8MX-afxo45DceC0OL9SPaU4Z31FYc4jRuvPj4cMt_Tw,44915
maxtext/utils/maxtext_utils.py,sha256=U2PVwfX7041S3e1f9QNb8QdElwUus9cmnNpKjDVjFOo,75480
maxtext/utils/maxtext_utils_nnx.py,sha256=ERFM4hZYBpTjRmdT6-ir0Rr95a238vVlthvGPXv6eNU,5670
maxtext/utils/model_creation_utils.py,sha256=CPHWFnTzgQl5a93Suz6vduNsTKHir5Ohyj1pts0sjyg,45116
maxtext/utils/muon_utils.py,sha256=KZyS1NkfTaTCNrRjL4tMRT-sCrOqysQ0ysGIc_AmW5s,6465
maxtext/utils/pipeline_utils.py,sha256=KFNKMqb-veMonRESwSBPPrcB6Txm7UZ0lAZJoW00N1E,16532
maxtext/utils/qk_clip_utils.py,sha256=MMpoEGqbyE-tYURPtElVHHXmKicHktCIgqVk-jo83WY,5651
maxtext/utils/rampup_batch.py,sha256=myxMMIk-5_de10aDPO-OKIdl_9VZwLQV6LnDxkvpCkw,4729
maxtext/utils/sharding.py,sha256=mJDwj2TKKPO1Ca6MkDbjGooG5urmFUyoiXkt6qIS5bY,28610
maxtext/utils/standalone_checkpointer.py,sha256=YToBoT3npkew9Yv8-_vdpZ-ZA-B1a7V9Hzg5ksAwrbQ,5409
maxtext/utils/standalone_dataloader.py,sha256=LskMPEYUTGCAXueNFYv_TtH0WJ7VGP0ugGVbs0yAqzI,2864
maxtext/utils/train_utils.py,sha256=wLhRDiL6InzPVdEbaQHIzPFx1N0GTlS1W7QJfuojCtQ,14984
maxtext/utils/vocabulary_tiling.py,sha256=hc-cDSpaZ3sZthDRoz5J65j6_XkDSXpLlsPVp8ZuCBI,10435
maxtext-0.2.2.dist-info/METADATA,sha256=H3qm_gbMj0vAyqC_DhnV--5kAOGiPkryTljP5XuwrUs,71515
maxtext-0.2.2.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
maxtext-0.2.2.dist-info/entry_points.txt,sha256=2SxQWEufgeg2foJdv5IQMooFfRQTxVX3Z5m4ULRPKYU,459
maxtext-0.2.2.dist-info/licenses/AUTHORS,sha256=kbdMM4KiM3YreLkMyk3BtUmt_wQpBOdL8q5wEw_Fa_w,12
maxtext-0.2.2.dist-info/licenses/LICENSE,sha256=PWvDbyxiaVFtjMjuY07bL-EkG5UZKAr0_lvdglV5UkE,11347
maxtext-0.2.2.dist-info/licenses/LICENSE_HEADER,sha256=sUtr7zKNxe6y-XzxLKv_5YHhGL__UFNYIHhMuwMi3s0,568
maxtext-0.2.2.dist-info/RECORD,,
