dependencies/dockerfiles/clean_py_env.Dockerfile,sha256=Rz-vglOzk6aoQQoLsQH8gcRGVTaJom5aYdb6BFfdbRg,2700
dependencies/dockerfiles/clean_py_env_gpu.Dockerfile,sha256=FZZUKq14f9HtsZ8-1Iapx1V6id5ADAAVG94FKqAZhII,3166
dependencies/dockerfiles/colocated_python.Dockerfile.template,sha256=YN906YJmFf3P052AmAxdoVxNURuuxXsYWsZwVoH0Ris,703
dependencies/dockerfiles/maxtext_gpu_dependencies.Dockerfile,sha256=4QmpG1r5grcfNZpg6Ukb3Pp5IehBdgymgWI-vfaLGKo,3268
dependencies/dockerfiles/maxtext_runner.Dockerfile,sha256=oMgEmCntCXgVFd1r5I3TdK2Y033G21b4XslpS8Ky_NY,611
dependencies/dockerfiles/maxtext_tpu_dependencies.Dockerfile,sha256=BRpHU5xTet32SJFwC18PXlXCi_LOFuJ39jsxrvbp1UA,3735
dependencies/extra_deps/__init__.py,sha256=3ZO2en3MVpSamgV0n5lAUOAuIKH3pqnakMBXyJAH1QQ,574
dependencies/extra_deps/cuda12_overrides.txt,sha256=wVNFe0Qxif19UzFI2TJVOv1eb3OnGKBYN-2k6TJYybM,74
dependencies/extra_deps/post_train_github_deps.txt,sha256=2cKxGHcBIO4Oar8TyRvBbx0AwtxfYgB-woAPxKE0XqU,304
dependencies/extra_deps/pre_train_github_deps.txt,sha256=sV1RDjptx-NEsGgkDRc8hmMnI62x9Z1In4XjopjXOD4,225
dependencies/extra_deps/tpu_overrides.txt,sha256=ghx-3IJn_a-UbCGKcCHAetvBe7OvcJXRLMGbobqycYA,74
dependencies/extra_deps/tpu_post_train_overrides.txt,sha256=YKiMiSkXHcVzKUMXZclt_viAyGRI_gKEb1QSJQRAbbA,180
dependencies/requirements/requirements.txt,sha256=dSbK348kxUuewZwvmnvKW5XU4frbo00DQs7zBlcc9Xk,824
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=CSDUl--wWdm253sjmZebKW1vVYrO1M_9N7-gd3JwbqQ,657
dependencies/requirements/base_requirements/runner-requirements.txt,sha256=MpoJLyt_GcqRiyLrk-hsn7zSlaYXRIl0o9gJ32mMMM4,71
dependencies/requirements/base_requirements/tpu-post-train-requirements.txt,sha256=viswTGibRFg7ImWLjbdYbL7WrgEXb3Tdm8e5XzkOgfE,472
dependencies/requirements/base_requirements/tpu-requirements.txt,sha256=aQdQVMbTv14BBLvF1yiV84DwUCFqdu6JeNxPv9Yvleg,20
dependencies/requirements/generated_requirements/cuda12-requirements.txt,sha256=16XpW6eR_7LbeqrgahYpyDQDUWqwTxw9JHeByUTre-Y,5948
dependencies/requirements/generated_requirements/decoupled-requirements.txt,sha256=I-Kel_rqaoCMcF2tSukN5a-mttSrla15ksk8TMsMhoQ,4591
dependencies/requirements/generated_requirements/runner-requirements.txt,sha256=OJ49BjQSadrzg3bvO1aTVBZ370djl7MCf2ZlpfVubuM,2828
dependencies/requirements/generated_requirements/tpu-post-train-requirements.txt,sha256=DnDhk1a74T2RiDQZu60kZEPr3AnHgHyB85PQJLPXlu4,7984
dependencies/requirements/generated_requirements/tpu-requirements.txt,sha256=IPtgVPOZCIWGU_SiDDr5StL1WCNqG_K46Jq2Uopck7g,5192
dependencies/scripts/build_maxtext_docker_image.py,sha256=6FQUAv2gkc6SEzNyUOElwTKqKYcvShB7xxo8ZFh1M0I,1170
dependencies/scripts/build_upload_colocated_python_image.sh,sha256=4GZ-3aeoIoumGCs7qpzvt0Csd7RprGL48pXLLsXIYeo,6579
dependencies/scripts/docker_build_dependency_image.sh,sha256=aYOkYf2qklJiCywTL2EAg3e2mh2cyVPFf9xu2UKhmWY,4560
dependencies/scripts/docker_upload_runner.sh,sha256=xucojKmS1tlzuR-zp9977Qik9RvmC0VvEf5fFB1TtDQ,6018
dependencies/scripts/generate_decoupled_requirements.py,sha256=TIewaWySu3W52MWGzgRKlu9Lp_xPChzOKuUuZ5fnQWQ,8651
dependencies/scripts/generate_requirements.sh,sha256=vcrxms1WlPtvVUI_PgRI34siL2Gha3l8RPbfErvt-wQ,7086
dependencies/scripts/install_post_train_extra_deps.py,sha256=zz6jcHjslVVrQkbPsvU3N6be8hte9eKJ1nlEJuU_piI,5434
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/run_tutorial.py,sha256=gftDfgkGiUtDWgTA3KZHav0gqlacjy-GYl9_bPJaXJI,1058
dependencies/scripts/setup.sh,sha256=0Dl7H1m7rzIGq_nuxZa7Ctqfd8tGIX3inlK15AaeDyA,15067
dependencies/scripts/setup_gcsfuse.sh,sha256=ZDeRzo6POsPHdFcAIECflL3qJXM_cQmsUCa3N4M5eAk,2382
dependencies/scripts/upload_maxtext_docker_image.py,sha256=237gMGAW0i2h3I3UPEFHBetORg_1LrdSr10QSjCr1W4,1152
maxtext/__init__.py,sha256=HKpzbyUK24LfPDWLLa7a8LwYp_nJd8Q0_1SQ9rM3swg,3882
maxtext/version.py,sha256=HlCa8mKjzXpxKUwkWUgRM10ijDYuNJ_AhjiHkI2npmY,883
maxtext/assets/tokenizers/tokenizer.default,sha256=mnsWXTocBQzJnDtjGgoiEeh9AvKu4dqB33m8ivYLei4,827743
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_gemma4.model,sha256=nzGKxNwC-FgOP2X_Czcobnw9Hl1mRze7LIyoErVFMWI,4689016
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/inspect_checkpoint.py,sha256=3AZh8YqdYPHiwcVrKM7KxOVtyAWGbxRzaJv2014vJNo,15984
maxtext/checkpoint_conversion/load_and_quantize_checkpoint.py,sha256=z3Jb2mGeeGjhXz8lhng9AAZwylEinEA3r0jGh3ZJeTg,1681
maxtext/checkpoint_conversion/reshard_checkpoint.py,sha256=mQQ8xBiYLFhpSG632Nf7qSS3fa4bKlUmbyBxYNQOg4E,6337
maxtext/checkpoint_conversion/to_huggingface.py,sha256=xhbq5UzDjuN3BAS01bF3hOZkQDkTXLfl-6jdDi3Csbc,24331
maxtext/checkpoint_conversion/to_maxtext.py,sha256=UQWMWo450LWroFoE4TyjQhwYlL_mY3WwVdolD9oAZ1k,49469
maxtext/checkpoint_conversion/examples/convert_gemma2_to_hf.sh,sha256=VAs5a_lOHMTJyJuOtuRV-hngPxaCGR1-k6ljN5yla8I,2926
maxtext/checkpoint_conversion/examples/convert_gemma2_to_mt.sh,sha256=98xQcKeK4hbSpGybCkhNd_xy94mYZMxCB46JhK2dUH0,2462
maxtext/checkpoint_conversion/examples/convert_gemma3_to_hf.sh,sha256=XNV_tiYJ3qSwMpL9pCsJR69h0RC32Q7jsyXggjsgqN0,2954
maxtext/checkpoint_conversion/standalone_scripts/convert_deepseek_family_ckpt.py,sha256=XCodZSoR2WZi6XIq7rgsLuuO5ADp2dJyEOiluuG0mnI,42414
maxtext/checkpoint_conversion/standalone_scripts/convert_deepseek_family_unscanned_ckpt.py,sha256=F233aLX0WiLNqbl-uyHRygjrYYgjp-p4GA3sfIN5x-4,17094
maxtext/checkpoint_conversion/standalone_scripts/convert_gemma2_chkpt.py,sha256=Pdun3X8MgbIyQUwBp3V0Zl289xHAQXi_mkqTRfWyK7k,12785
maxtext/checkpoint_conversion/standalone_scripts/convert_gemma3_chkpt.py,sha256=-hlnnZJgorVBirqxZymS-9-WBgM0e6KPB5k9ACUuOpk,10264
maxtext/checkpoint_conversion/standalone_scripts/convert_gemma_chkpt.py,sha256=NKoqsFI5tuQsG7TYr7EVhsajPQVRtDy7sQRQFIRqII8,7191
maxtext/checkpoint_conversion/standalone_scripts/convert_gpt3_ckpt_from_paxml.py,sha256=Hf5Zy2ccX8CE_sOqneIEFhlqduJNW2oA9ipC81KQ97w,13483
maxtext/checkpoint_conversion/standalone_scripts/convert_gpt_oss_ckpt.py,sha256=A1YBcRUdN2sGCDT8jfRwTJnjN2zj1e9kXZnSv7-N7-8,17869
maxtext/checkpoint_conversion/standalone_scripts/convert_gpt_oss_unscanned_ckpt.py,sha256=DVWkFWAwjVe1Pe-YGjS3tSUUBgvyL5mhrt5d3mdZHdU,15221
maxtext/checkpoint_conversion/standalone_scripts/convert_qwen3_moe.py,sha256=EGNkI0TOuDyknbptgZXY2mN3eIMwFrGHwFw_ocVUYYw,13504
maxtext/checkpoint_conversion/standalone_scripts/convert_qwen3_next_scanned.py,sha256=MEtIIFoG4bC9H5viTYfanKLr9a5P9x9m-1UpWv1kOpo,21643
maxtext/checkpoint_conversion/standalone_scripts/convert_qwen3_next_unscanned.py,sha256=vWzuIZB0ZMEo87TDzNXT8bbGrSW77YAlSBYEpISLYGU,14059
maxtext/checkpoint_conversion/standalone_scripts/deepseek_dequantize.py,sha256=Gy1183A2-1xWmQyeQhEI7SnWD9TVDNXArfydkRrLFnM,8798
maxtext/checkpoint_conversion/standalone_scripts/dequantize_mxfp4.py,sha256=RoCBBfsf_HDgwmODLiUP59H8us0aGK8f4-sXzFxxNy4,7069
maxtext/checkpoint_conversion/standalone_scripts/llama4_ckpt_unscanned.py,sha256=qL_zCdgsnbp8V3ExhYS7jalVTYO7kHVOxGVVYM3SAqo,43818
maxtext/checkpoint_conversion/standalone_scripts/llama_ckpt_conversion_inference_only.py,sha256=MpxXR5t_kIfDYg_e57i0C3d6yNDfFAVAUDbKSQ10XFg,12256
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=xUDYMaFwnCzIRY_TZb-ozHZBPLY38i2JdIcAzM_-2r8,75465
maxtext/checkpoint_conversion/utils/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/checkpoint_conversion/utils/hf_model_configs.py,sha256=9EPPH9UkOnRS2V6whG5pNcqdjbe1Mle56CQuRgUVfk8,57976
maxtext/checkpoint_conversion/utils/hf_shape.py,sha256=IYSJ2fz5qTUywC6Y7os301ebGaIaAEuaXB3T_uhYm48,59586
maxtext/checkpoint_conversion/utils/hf_utils.py,sha256=TcwALFP7YqStfBHw1areoifc09rfrHIh216Ok3_eCus,8432
maxtext/checkpoint_conversion/utils/load_dynamic.py,sha256=qdqSr1w71raop0xucY9UTrhbGznjXRbCy-6mpwlS8JA,14438
maxtext/checkpoint_conversion/utils/param_mapping.py,sha256=AjCAe0YfkKJGXdJnudpVMCfg8wI-OYBhFBajcRrDoSo,207591
maxtext/checkpoint_conversion/utils/tensor_handling.py,sha256=2m-bFmaZN3CRb6uw5so-186nP_mUX4aagPFX0OUDZDo,7408
maxtext/checkpoint_conversion/utils/utils.py,sha256=0T6GkWdWZtSMg2qgpo5eIOuI4F2aGfZBKPDsPLmD6ps,56582
maxtext/common/checkpoint_context.py,sha256=e6glOc1V7AosXcHB_tFpHj0gcP1I6meYZG0ywVSBeU0,7368
maxtext/common/checkpointing.py,sha256=cz59TKMlifw_tLDVQecZoxmHT1jp0ReUevGLt6FxPjM,46347
maxtext/common/common_types.py,sha256=nnWG4DAnAHd9oXFkIyxO3iECvFrGc0e2zFn8PXxgtxc,4507
maxtext/common/data_loader.py,sha256=lCk3aik_EM6Q2dLPm4h6KC5exb9sZY5psDcLzP8PJTk,7140
maxtext/common/emergency_checkpointing.py,sha256=UTF9_Ar4DMltWGn5aYIU7TBJyxpQvMVk8TVklV9-l4k,7561
maxtext/common/gcloud_stub.py,sha256=8Ig6V8-F_wV1RHk0PPhgrakaTTVTZTTwhADOYzxRBoE,19368
maxtext/common/gcp_workload_monitor.py,sha256=9pPuT4UzQABwQrrNiOCefUZ1mLptFMIouMNSui-gdaw,9117
maxtext/common/goodput.py,sha256=521KGcpifrDFsOJU-bWIC6_NoMl_NPABe_zEKw_gO-8,6576
maxtext/common/grain_utility.py,sha256=5bA8JHZdia8IXY6R4i8iSpgoVSvv5qr_e1zEcYtmV54,25158
maxtext/common/managed_mldiagnostics.py,sha256=bH_MfDcADVRway_zXHQmFc8wq9dnWkcOuSUDTPvW_dA,2662
maxtext/common/metric_logger.py,sha256=jmm9N5LBFh87CTBBS1ePwrmkiqTCBt-40ESMCpfMPbA,23748
maxtext/common/profiler.py,sha256=Ohl--PNuaHcIpAd6__vf9ClVcnEXDXZfv_ynsuxvngY,7337
maxtext/common/train_state_nnx.py,sha256=UH01UZzAtDt037ozEPtBwVWP73W_WozbbKTMk-3NFbE,9591
maxtext/common/vertex_tensorboard.py,sha256=KULpw-MR-yllPB9nECp4j-fwtlEWHHux6jSUFmQF2Bw,5267
maxtext/configs/README.md,sha256=BiXZPp9dBip-hfTrX2B9MTylxsKBHu1UKbGbsWDJ0vA,32172
maxtext/configs/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/configs/base.yml,sha256=Z_s3yTkNvhu-kIAtIxZhnn759JRv8c_yu6kdK2Ip44o,78110
maxtext/configs/decoupled_base_test.yml,sha256=vtTS0h5T8mNm1Itbrlae3SdvhbAeAcH3o38ddJ7ffdA,2375
maxtext/configs/pyconfig.py,sha256=ecTGTjNDIkq9HsYkpu88Kz0btSzzUfEbgLq9GVvDxFE,25117
maxtext/configs/pyconfig_deprecated.py,sha256=7QBYMewboIg5Z65DJrs2SuCLnyOaSJYGYLN5FjzYL9s,62073
maxtext/configs/types.py,sha256=ajGlFR5Z5GhOCQ8fWJWkFLUlyD2m1lUFTmJE9E_vRaY,219255
maxtext/configs/custom_mesh_and_rule/cp-as-ep.yml,sha256=qFqXqn9nsMwvUY_KnKFejCMRslZ3jgG9U5ODBz6WAXk,3993
maxtext/configs/custom_mesh_and_rule/ep-as-cp.yml,sha256=7Q5LWF2S7WTpiKG_ZTngZ2nu1m1oYmDEPQGVhSkP4wI,3849
maxtext/configs/custom_mesh_and_rule/ep-as-dp.yml,sha256=iN8iYtrFZVgTaxwyAOvjmFVCZ1pIRNcpdSkPjO-zV94,4463
maxtext/configs/custom_mesh_and_rule/pipeline-large-moe.yml,sha256=Q-fTFJNJIbySi_PuH5qVEAjqojoG0jdyRgIFp-5zJ1E,3983
maxtext/configs/custom_mesh_and_rule/pure-fsdp.yml,sha256=jx1EOfjtaE-2S6V1XLGOO1l0j_Wk-lqS_UPkk3UH2W4,1584
maxtext/configs/custom_mesh_and_rule/shard-exp-on-fsdp.yml,sha256=l4bJ7LE-KBqxv2Vmwhi5uc9OYCWxF_Uq6zdbnVGejWE,4167
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=-SLRICaVHsIjg1W2K1ZgSsX2XbBda_nDD3DXsln8lEs,1676
maxtext/configs/gpu/a3/llama_2_7b/1vm.sh,sha256=Uagwk6Or-i55kcarapEd9aX_2-p4KolN0aKuiBJZGFk,1651
maxtext/configs/gpu/a3/llama_2_7b/2vm.sh,sha256=w3aYJPDbKbL7ybrdAvlJkVEbpSND85f8asbsUAnpoy8,1620
maxtext/configs/gpu/a3/llama_2_7b/4vm.sh,sha256=dO0P0bMNm0PBMRL45njsrtckZTmyUx07-T3svZUKrtU,1620
maxtext/configs/gpu/a3/llama_2_7b/8vm.sh,sha256=y-Cfhzr5aPqZb8sFwYjaclB-RCuvO0-fqwQc75S5cwE,1621
maxtext/configs/gpu/a3/llama_2_7b/README.md,sha256=Vu3lkSSbtK_Yt7NCitOMdDfPEM0xJfva5LaooGeJEUM,1042
maxtext/configs/gpu/a3/llama_3.1_405b/128vm.sh,sha256=bDkvzt2fCEWnGpEYQS3CT64JR4Pi2p--BdPDI2sxbTA,2336
maxtext/configs/gpu/models/llama2_70b.yml,sha256=DsDkr1wMEkX6wicmHRZkNgDDAplEo-gfleQ2_AmL5dM,442
maxtext/configs/gpu/models/llama2_7b.yml,sha256=llVAOilborOAOar9v9oaLxH5nRKOIyvkiBFGI4DUzvk,362
maxtext/configs/gpu/models/llama3.1_405b.yml,sha256=6CduCFh1tCUC1J07gxEaLOxYfByJuJCms5LFWqtfHQs,446
maxtext/configs/gpu/models/llama3_70b.yml,sha256=GCgrpMIwOBrMmP5036amiliMp9P1iSFXSYvnVkc8p_I,1052
maxtext/configs/gpu/models/llama3_8b.yml,sha256=QZ-2wbcEgMQwNzElatgZc03QTovuI4f8Q-ElsMgk-Zs,1067
maxtext/configs/gpu/models/mixtral_8x1b.yml,sha256=Ove1LPIkoi62u2H3sOdg52E3d8GKlOm_bPdcccy4JjM,1144
maxtext/configs/gpu/models/mixtral_8x2b.yml,sha256=eiPDQp6SMNetAdyLpSn1_ApFw0haXiEHvNNWdOWOBn0,1145
maxtext/configs/gpu/models/mixtral_8x7b.yml,sha256=RncfMNANkqKE-xDpI3J0gaLgjTvjusSm_LCIfVi2pt4,1221
maxtext/configs/inference/inference.yml,sha256=AKsWqun36sphCri5DstyuREo3WfFQCdsEtdDn8MedCo,5034
maxtext/configs/inference/inference_jetstream.yml,sha256=PSEq1XjSQLIIiKLSMxcGR9YsIhcw9fB7DsPQYH0jJmw,104
maxtext/configs/inference/vllm.yml,sha256=EukYCEd6H3gGZ2xcatlYxvBgsfQD5lIgqT0nKH0cZlw,5702
maxtext/configs/inference/multihost/disaggregation/llama3_405b_v6e-16-16.yml,sha256=RxAFOr1ISi41MlqagfVOV6y1hj7YwXdIH-aEgiDyecs,1753
maxtext/configs/inference/multihost/interleaved/llama2_70b_v5e-16.yml,sha256=x7aPWwwF6Gp3TBRQN7pMuQ7fXQNjHEuUHG2aaj3Tdho,1932
maxtext/configs/inference/multihost/interleaved/llama3_405b_v5e-64.yml,sha256=Oq5egO6Vdmi4_1-l5jtXAJxYzWbtay0U5uL5QcCa3GM,1805
maxtext/configs/inference/multihost/interleaved/llama3_70b_v5e-16.yml,sha256=As4zgDs3t_dNX82xOBkQ5jqJPFgThelMrQ-T1UeapiY,1983
maxtext/configs/models/deepseek-custom.yml,sha256=8C7CmfLrLyGeHAUXgr4tdlDOF8D2Biwm_Z4iX3oqTxU,2888
maxtext/configs/models/deepseek2-16b.yml,sha256=K0wus8WG0oyt0W9mX0E-aZG4g_E2-cWwLoMT3zCbkNo,1482
maxtext/configs/models/deepseek2-236b.yml,sha256=GfTH6Dp7EtZ-6z6MRAkEqDOjTXpZFK0rGq2LZuoVUMs,1555
maxtext/configs/models/deepseek3-671b-2dfsdp.yml,sha256=AQWpnR1aiqLTEd5PhragR_3R1c6c_bIgtSh083Dz-dc,3127
maxtext/configs/models/deepseek3-671b-batchsplit.yml,sha256=B_d84mVSK7kl-TfAebywoOK1iwcXHy20r0QC5VTPPNE,2786
maxtext/configs/models/deepseek3-671b.yml,sha256=N68HLAyfGXx3O7ve17P9CrSV91HYTna76Py3QYKJBrI,1618
maxtext/configs/models/deepseek3-test.yml,sha256=68ftDOajNfENrPx3Suc8PKBB69uPZKL3QHhzb5h104Y,1515
maxtext/configs/models/deepseek3-tiny.yml,sha256=_0J4c0DhGdWDmHqJVzUWd6griW-yLTJmiOFpg2bU2GA,1399
maxtext/configs/models/deepseek3.2-671b.yml,sha256=8XgaYtIRxmeB7v46xi0LlUWZmG4Ef8b8ZveJd8YQq0M,1670
maxtext/configs/models/deepseek4-284b.yml,sha256=P6g4aAe26ZzbHD9cixBfrsjfQf7MGMqo_pzY7YnM4OE,2369
maxtext/configs/models/deepseek4-tiny.yml,sha256=6dWx1-KvJCb35iwm0tImeRCPRpByMXGzXZ_FEbzzrVM,2230
maxtext/configs/models/envy-switch-base.yml,sha256=-Hoi_R7Lz1ppyfCFts1M1Rt5kKsS_-cX_Ri_2VbWW34,1159
maxtext/configs/models/envy-switch-large.yml,sha256=O0L_Y6y42tI5wopeZuQbpX7pKAUQc4JyieVgaPK9y2w,1161
maxtext/configs/models/envy-switch-xxl.yml,sha256=5HcIRUKNJZzIJ728RUVX0LLJflZuyakink3ZrgHJwIU,1159
maxtext/configs/models/envy-test.yml,sha256=f_Poq-_9pzLkqllQH6juuN8gvKtjEoKv8nGQ3BH4x1o,1137
maxtext/configs/models/gemma-2b.yml,sha256=P5jzBL2USmfm54IAN8cIeaMHlfqB71kGkmfbJ9z1W4o,877
maxtext/configs/models/gemma-7b.yml,sha256=ikUVQMttfO0e4MWvySxZFRXuSDLVwZ1f0EN6hYRD_rc,879
maxtext/configs/models/gemma2-27b.yml,sha256=knYcEF7KZerkX3xee3HhYS7p-iR_mLltD68Fij2lbFQ,1116
maxtext/configs/models/gemma2-2b.yml,sha256=acDx5bXr-1LwqiVc89Yht7nRQUQSTgpQcnFQncTNWhc,1111
maxtext/configs/models/gemma2-9b.yml,sha256=qfDhp_M94e2SY7MAOK2Pyy1tiaMRlAvUX-WtCKp3jb8,1114
maxtext/configs/models/gemma3-12b.yml,sha256=fZbWxp0RwxhJoCpZ4ItLUhQAO71X2lftwC2F1QRffpM,1386
maxtext/configs/models/gemma3-27b.yml,sha256=bRir9U2AnBYKUfr_DN0alQ1olMSsl3x-mf_95KWsVgA,1387
maxtext/configs/models/gemma3-4b.yml,sha256=FR9BRiPLgv1W0WDO_x41xqAfAEvwhR_PcwbFvmMBUEU,1384
maxtext/configs/models/gemma4-26b.yml,sha256=TT1mLOpFvsMZR0HP5ZbVaTcEjIgY3msM_K0syYxwako,1813
maxtext/configs/models/gemma4-31b.yml,sha256=WVo9ePQcbkV2XdEq2XyKwvQxeIEGoamkF81a6GSVZIo,1692
maxtext/configs/models/gemma4-e2b.yml,sha256=OjKUeguUm_G-XOiNrnVpMlbXgWkJWk9G4B5XZO2fMtU,1835
maxtext/configs/models/gemma4-e4b.yml,sha256=d7A6_3m2kB1Su0bde3PbktN9dtjy3vSJi1_J_8PYO9M,1858
maxtext/configs/models/gpt-oss-120b.yml,sha256=UaRp0RPgv0dlQYeNPuhB93KvKvpMZOy7yHbHovpElj0,1502
maxtext/configs/models/gpt-oss-20b.yml,sha256=iKY9VdVXkkZuP0OK0seG5lpcQ4H5QJTwV7QsDXH_Umw,1499
maxtext/configs/models/gpt3-175b.yml,sha256=uK8fHmKOdCS99f6JaPpGGXA8d_Jb0PJyXEK18hFeYds,1321
maxtext/configs/models/gpt3-22b.yml,sha256=iIYjO73qmIrenLbyrYwBRvhMWmFPV12ACy6ILHyjdZw,1161
maxtext/configs/models/gpt3-52k.yml,sha256=i9dRmSyoT_70dB2ul1JQp7u5fBDFxkuSAGRXX5_qw98,1258
maxtext/configs/models/gpt3-6b.yml,sha256=dJc3VAdFHknFFOb-eRJ8BaO0AdOh82mvESEb39yAbas,1160
maxtext/configs/models/kimi-k2-1t.yml,sha256=2nkWgdy3A288-OeP4T2ZR8bneWdFJB0qh4UBQ6Rkesg,1530
maxtext/configs/models/llama2-13b.yml,sha256=985ifJftf2MTcphu112Ci7x6XjqsG8EJ87gs7xQ9_18,943
maxtext/configs/models/llama2-70b.yml,sha256=NJc3l8Cmptw8C_JcEvXmYQPQu1249rz5bxFzUPlKq_o,919
maxtext/configs/models/llama2-7b.yml,sha256=KfCVZzro4OzId1EQNNEojKcADISjAIquG8l4m-SEVjM,942
maxtext/configs/models/llama3-405b.yml,sha256=AOd7utGRHIvS2YoHlWhtdRYBElZAvvs7C3P5aoBRl6Q,977
maxtext/configs/models/llama3-70b.yml,sha256=FWGlwoHktfQYat5tUe9sa9eAF6K2AYjChhCFc-tNIuY,972
maxtext/configs/models/llama3-8b.yml,sha256=6GU6hQggvytVEA4dwSc25zZ38eaVNWODwvJ4Sj1OZBg,971
maxtext/configs/models/llama3.1-405b.yml,sha256=OpoDqvnYbomUXfH3ERkzFFAoAzjHHvL3OU-XNs67iw8,978
maxtext/configs/models/llama3.1-70b.yml,sha256=AtyCECX1NMZNJkrKl29EaB8eaJTAPPoftnVw3UpOumM,974
maxtext/configs/models/llama3.1-8b.yml,sha256=ub-D8qto_xPdVe7VWYi8nWIiudE1xp_syoPzCZ1PZ_c,973
maxtext/configs/models/llama3.3-70b.yml,sha256=6od89hmR84TNS4Bp9a_3UMZgeaqxK5kITCCEI2Ecqm4,974
maxtext/configs/models/llama4-17b-128e.yml,sha256=5xFvmSRAwXQdkJIGe_WVtra5T-bgekR0qqUygmR-KkA,1796
maxtext/configs/models/llama4-17b-16e.yml,sha256=G2RANXrgjMGnAd8u9Ots8HHJ-Rdno-IMqEerp0UZxO4,1830
maxtext/configs/models/mistral-7b.yml,sha256=TE1Aqs0cRi-oiNzwAY6O_YarzeSNM8MYA8LR2EiQBzU,934
maxtext/configs/models/mixtral-8x22b.yml,sha256=c35L3k6SQY99HQzqiElXJj06GXj_FDX_IOqRsTjqVnk,1058
maxtext/configs/models/mixtral-8x7b.yml,sha256=I53ZY6omUzro3WQJ2uJaf407Wy6S3cvB75xBU7ST-wQ,1057
maxtext/configs/models/olmo3-32b.yml,sha256=weNAT56RG8dm-Kfm-EJMMMe6GNTCZagsDXqXNod2zQY,1731
maxtext/configs/models/olmo3-7b-pt.yml,sha256=iuGkIsJ_8Ce0KOsqXohFpJDKWjO3qgOdTh5ZLTVFVj0,1874
maxtext/configs/models/olmo3-7b.yml,sha256=RkHirb8bnsEvxQvRtIycxODNrE7fCX3ZSQjNqKHzpVA,1715
maxtext/configs/models/qwen2.5-1.5b.yml,sha256=wfUwXaMITpBxXXOvMs1Cs-y64oscU_5Pc7wmShQcfvc,1099
maxtext/configs/models/qwen2.5-14b.yml,sha256=046YBkIkpukdJUVOUbaYoexLjtmlMoXwvyZOzuSXtR4,1100
maxtext/configs/models/qwen2.5-7b.yml,sha256=VECwVjZAy_4gIvkDZ91meabiJitSZj3oaHp-pRymJXQ,1097
maxtext/configs/models/qwen3-0.6b.yml,sha256=0KD5ljFYgMYy60OCZwAxmHuybsXvtx6BMqKI_FL9ZGw,1118
maxtext/configs/models/qwen3-1.7b-base.yml,sha256=zuvq-JcFSCu8CXqTvnpDOo_ZtUxKK_Wsxmm3AmfqWxc,1123
maxtext/configs/models/qwen3-1.7b.yml,sha256=17HysIrgfE6BORyIkX-KQGx288MpLeJJkS6TLqlKaUw,1118
maxtext/configs/models/qwen3-14b-base.yml,sha256=nYdtbeTcd-BjqEoUoyzN7Lsj0i8VjXRwvLkcSM-9ciU,1107
maxtext/configs/models/qwen3-14b.yml,sha256=7J8qGkfR2Jyd6DZwqopkInKVA9SxFlUHQocbPQC3LLk,1102
maxtext/configs/models/qwen3-235b-a22b.yml,sha256=dzHLQLB6ZWnv0B-EHE0y3DwLGtXz_aVdHQOigETzObQ,1143
maxtext/configs/models/qwen3-30b-a3b-base.yml,sha256=1VWETtkNBR-SsEEZ9lffDW34SN-2qO79rxymQgmoihY,1114
maxtext/configs/models/qwen3-30b-a3b.yml,sha256=G1pkc0hUCIcpnhivDNR5h-nFIaI2jbUDmTlmFMRvdB0,1110
maxtext/configs/models/qwen3-32b.yml,sha256=F76oBljIgpC8RmTY3LhrFP22IPYECy8MUPYlJW0_IVY,1102
maxtext/configs/models/qwen3-480b-a35b.yml,sha256=jo5XNsb0aPDSdrxiKbni7lcYtlZad0UfkmO50zk951g,1129
maxtext/configs/models/qwen3-4b-base.yml,sha256=HRIFGijzx8pGHL71giruo9i3cVRUmgSGRIxV7RpLPB8,1121
maxtext/configs/models/qwen3-4b-thinking-2507.yml,sha256=o1psry5eeECQgd3BVLBWO8PwFq4qS8MrLekKNPIPaY8,1100
maxtext/configs/models/qwen3-4b.yml,sha256=xLLBlHKgM9cG0m5Yat4MIymsNbf0_6Z4dFxWZm2t7DY,1116
maxtext/configs/models/qwen3-8b-base.yml,sha256=toFCoeftIBmZNBQhOWPyG6rPVlxLymMC79DErKEYFAM,1153
maxtext/configs/models/qwen3-8b.yml,sha256=CoyCQlJE1YPGoSaquXzC_te_62BXmjfuJphhNhDtig8,1148
maxtext/configs/models/qwen3-custom-30b-a3b.yml,sha256=ddVw3kooa_qlT0V4nniESFmYEo-Kjrxx8HwfN2f_NrY,1185
maxtext/configs/models/qwen3-next-80b-a3b.yml,sha256=3l2dufNOUpFlEX6yxSZY_FVoS1ZOPQwBj_JjR_m07cE,1497
maxtext/configs/models/qwen3-omni-30b-a3b.yml,sha256=qCdk2_6U37nYb84Ga5Vbfl7rkUEodBlkrgpcFL1V-Y4,2610
maxtext/configs/models/qwen3-vl-2b.yml,sha256=p12tca23B3e-TTCFWs0UfgmrA6khmrdTkmIVnzoktcM,1645
maxtext/configs/models/qwen3-vl-30b-a3b.yml,sha256=I6XQq3GB6lyu4fMkkFJPQASjxnDfDuvMg09i8aGyRi0,1700
maxtext/configs/models/qwen3-vl-4b.yml,sha256=DC8HK9BOlwE3jaQS3ndlr65tc7EfsZDczr1KCll5lN4,1645
maxtext/configs/models/qwen3.5-35b-a3b.yml,sha256=5ynJnkTXXyNRwgrl_R5qQF1eduktUCcLa3pJXdq8JTY,2081
maxtext/configs/models/qwen3.5-397b-a17b.yml,sha256=lYkEB6XfVBrgvBMmAUjZ7YI8o8fO0GKalthE-ahLnJI,2131
maxtext/configs/post_train/distillation-sft.yml,sha256=B89QGdKUKQvH1iLUjcYHHwcGA6SYtIPSjx4_E9Ij6KA,1561
maxtext/configs/post_train/distillation.yml,sha256=l7DEKpka5f-kemYgTnuKXSNljPzq9X9xX_V-U4ZuX_A,1662
maxtext/configs/post_train/distillation_gpt_oss_20b.yml,sha256=0z1QMgYsGyQZ6WL7MRY_Jj8zT6oERZbx8xzq52-fk2w,2942
maxtext/configs/post_train/distillation_qwen3_30b_base.yml,sha256=aTMqpT41-DU5DSUUjHgYtmzo6IDYcD1gT5ugl07apgY,3012
maxtext/configs/post_train/distillation_qwen3_30b_base_pdbs8.yml,sha256=PM-GYXcwl-_EMtlDcWRaYWuWrFN9SdMauHXTnimR568,2864
maxtext/configs/post_train/dpo.yml,sha256=CzG33QEef_yeArDeUgSPkvkDG6JCrnJ695AkJUz6cVQ,792
maxtext/configs/post_train/lora_module_path.yml,sha256=0ji-cumhxO9Wj5nKMTbwE6YPo1UprJmikPZA-j0EhS8,1957
maxtext/configs/post_train/rl.yml,sha256=PkmUnXtESaUPGbnelPZE36BCXfKDp_fbqhJ_AdkB7NA,13055
maxtext/configs/post_train/rl_mt_jt.yml,sha256=2wF_o6PGUnmy0oU2-3M9Y09o4z9QOp3jo01YTKJBkFc,5267
maxtext/configs/post_train/sft-vision-chartqa.yml,sha256=tdcTVmJFkQAE8IRrKfJyIGKtfT55wRl3KaX0C2NPu8Y,1226
maxtext/configs/post_train/sft-vision-coco-captions.yml,sha256=-v89C1L8jYf3Yb7SCwpsQYGDV0DZu5MnbdN3mJIsRIw,1440
maxtext/configs/post_train/sft-vision-slidevqa.yml,sha256=emEMk3tz9tu28UlbYBEKhPO1l67NYaCvCoHloqDi3K4,1750
maxtext/configs/post_train/sft.yml,sha256=TJxIxr7tP6kmCakqJMV1UDGzhk_fkmBjUQieEOjJH3o,1454
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=5a0NxSWQbiPWbgKCRbzEuKD9fq6M2TWmGlFWfKhRAIQ,2348
maxtext/configs/tpu/v4/52b.sh,sha256=SO5ThQ-HY_DSFtrHSTesWriDl5hE-2CedrV7QGeoZkI,2412
maxtext/configs/tpu/v4/README.md,sha256=MA1qH_XaDpC12OIDbe06SEKs-8EeZGi4HjQyoxV7Aqo,1292
maxtext/configs/tpu/v5e/128b.sh,sha256=lfr3KglObEER2O8Tjux0E_qYdf4Bv0M_aw1Yled7LrI,2140
maxtext/configs/tpu/v5e/16b.sh,sha256=RmlDIqYBS_LMB24zgH2CIbIXb97Zo85MnutjqzPai4I,2026
maxtext/configs/tpu/v5e/32b.sh,sha256=0Oe7C6hq5EJTblnCvGqszmZC50Dbf_TNKcXjXlVIaYI,2025
maxtext/configs/tpu/v5e/64b.sh,sha256=hglSy-M8lfvVf3aXzaZ-rPzaIh5wXE1nU6aaX4kijY0,2025
maxtext/configs/tpu/v5e/README.md,sha256=nVDl97pgUVNIG3Sj-uK4PKRllbXqmx-4fcY2n-dKTqQ,1878
maxtext/configs/tpu/v5e/gpt3_175b.sh,sha256=KXAQzZtmCCAPEwrc72ZNGrRWXWGty0qFdc1AX2xHsOM,2075
maxtext/configs/tpu/v5e/llama2_13b.sh,sha256=Rl15A0hIaI5th4EP5dNr4sG42xygnjWrs8ATDAtHvE4,2064
maxtext/configs/tpu/v5e/llama2_70b.sh,sha256=ShV6a3KDb0r-jCf7WGU9mp3HM1KJ7bfBvZdViD0XPmA,2064
maxtext/configs/tpu/v5e/llama2_70b_v5e-16.yml,sha256=x7aPWwwF6Gp3TBRQN7pMuQ7fXQNjHEuUHG2aaj3Tdho,1932
maxtext/configs/tpu/v5e/llama2_7b.sh,sha256=aF1uPh8RvJqCul4V38cBMtOUZoX9ZuBdYQkNdY7uFMQ,2054
maxtext/configs/tpu/v5e/llama3_405b_v5e-64.yml,sha256=Oq5egO6Vdmi4_1-l5jtXAJxYzWbtay0U5uL5QcCa3GM,1805
maxtext/configs/tpu/v5e/llama3_70b_v5e-16.yml,sha256=As4zgDs3t_dNX82xOBkQ5jqJPFgThelMrQ-T1UeapiY,1983
maxtext/configs/tpu/v5p/1024b.sh,sha256=yHA9RYniHC--g9DgP7J5C2RR1oSDoMljhLZapumDRmQ,2249
maxtext/configs/tpu/v5p/128b.sh,sha256=-4R_OdpxixS84HmDeoKefILiTH3JR8c_OreTs9WhuoY,2243
maxtext/configs/tpu/v5p/256b.sh,sha256=Oa3EmqKTc9YiAqfIseH9KY26956ZSfDTcsxcG6SqPss,2297
maxtext/configs/tpu/v5p/32b.sh,sha256=827DkBS_0PsIP493YbU1lCzX0X-smJCnvuIW3pcyAJg,2439
maxtext/configs/tpu/v5p/512b.sh,sha256=zJ6KbD7fPaAEZQuWxNU1hxH7kZXNdc-dmtPcrN4-f-0,2294
maxtext/configs/tpu/v5p/64b.sh,sha256=clKMShqze1BcxaBy3H2zRbknrSX8fuBlClFwWWopQ7Y,2227
maxtext/configs/tpu/v5p/README.md,sha256=ydRcAfq3iBZZQ4DDTBOZYTuNYWUB6WQqPEujmRKXlFs,2167
maxtext/configs/tpu/v5p/llama2_70b.sh,sha256=vV_1MaRO-lh3upaAfNv8ivozk0NQBryAN32HehjrrfE,2558
maxtext/configs/tpu/v5p/llama2_7b.sh,sha256=OWDLgYvPqw4gJ-XomRNKfu6GfQuyzlI47lZVK4OVeXg,2596
maxtext/configs/tpu/v5p/gpt3_175b/gpt3_175b_base.sh,sha256=TcywYh5gkYhrIYL1zUgFsPRp4q5AKrwgFcm8TmFTUWY,3948
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=dzSxwtXfxvNu7zaBDh-gv_CTpZlmYcClQh-Km14IPGY,1992
maxtext/configs/tpu/v6e/gemma2_9b.sh,sha256=M0EwGTcpjRPQFo-cvq2dQaDQJzJYCYwawLjp0gA4S4M,1988
maxtext/configs/tpu/v6e/gemma3_27b.sh,sha256=05YENBp2pwaOGlnVXy-dfBbxcElZJsXVkakus7jis3w,1992
maxtext/configs/tpu/v6e/gpt3_175b.sh,sha256=ZpejqLI4v2eFxPFPAyKIJGUzvm32-aKPta8Tm6Nizb0,2005
maxtext/configs/tpu/v6e/llama2_7b_4096.sh,sha256=biRW0Nj5kq_SmoJgFcvUQkM-N8lsShhWzcnwqkXAX6s,1842
maxtext/configs/tpu/v6e/mixtral_8x7b.sh,sha256=nqYWCIxE1hV8DbcK3mQWqJwK9_83fwUxcL7xmXrBypk,1846
maxtext/configs/tpu/v6e/inference/llama4_maverick_v6e-64.yml,sha256=291jskGxgIwNPA3BhsT2NPTcnHnJm7CQ6BVv9ZA1KHA,2757
maxtext/diffusion/__init__.py,sha256=Cm4xRiz4DtammIr44VQW43aaKB_yHffADqjWunnrhw4,631
maxtext/diffusion/block_diffusion/README.md,sha256=j3kxoIkq26Y9GG1IaFb7LcCXy4UxfQ2nJ0jGDYk9Wq4,2062
maxtext/diffusion/block_diffusion/__init__.py,sha256=wsVBaR2RkQUXsuhI8SPop4mpUjuJnTB5LsU7bUzw8fs,627
maxtext/diffusion/block_diffusion/corruption.py,sha256=uuXVGz9SjgOs8O9il2Ut388GyVRKofMj50Rvb4L9oV8,4688
maxtext/diffusion/block_diffusion/target_alignment.py,sha256=5boXQgMFHXzUtbLgoQc9smeDe8IRxiPzXCcg0dQnCto,2524
maxtext/eval/README.md,sha256=k1X7m7fVUI3gskAYAI0Xb5nL3shVo-xG5d5eKsSjoO0,8101
maxtext/eval/configs/base_eval.yml,sha256=W9Sret-EmVr332eBQJ84ddTePKhlgLCGG7zJaSLGqKs,152
maxtext/eval/configs/mlperf.yml,sha256=fr19OS3zECY0pT23IyEVBTqA1gb5vGWB5JkQoCOva4c,102
maxtext/eval/datasets/base.py,sha256=jcESEA4SHehIHGg7OlmR9Mh5d7UQ-PgFbf2LKSAStMQ,1789
maxtext/eval/datasets/mlperf.py,sha256=eu7BvsCKz9b5RvwXHqa6OqTGsmK5DGAn0RHgd00emJM,2049
maxtext/eval/datasets/registry.py,sha256=1sKZzSzr_Ag2Q_cOqMTJoO5P5L8OYcGGm4GNJiXr3es,2045
maxtext/eval/reporting/gcs_reporter.py,sha256=zPsWauJ1C3N_pE6lclXGptl1sjZSm1d-lbeZQr5Kuuw,1226
maxtext/eval/reporting/json_reporter.py,sha256=R1EHdCz-VaAG8N-VT0J_IbMcZvni6ZgFFA9CNN_TKm4,2833
maxtext/eval/reporting/simple_evals_debug_reporter.py,sha256=iB-T97B3ksOdbjBLsZLTlwvRZn69opYpSTdwcHHc7ac,14271
maxtext/eval/runner/async_client.py,sha256=HVYbm_qFOHaKhUNkwylsZvoh3drMJIRLj-eMEHe6shU,5566
maxtext/eval/runner/common.py,sha256=_4rb084_gIW5gfr5B_6CY6i6EqcuGnDPpMKP4aTtfyc,5962
maxtext/eval/runner/debug_utils.py,sha256=jPGyMTOHE_wolgmmAwPjkRIm_ERyk-2ceQsdAgDNvG4,4355
maxtext/eval/runner/eval_runner.py,sha256=0gcNckV6d-sWyIfe8RVdZcWuw8b7sBokkKLbu7_VXaA,10406
maxtext/eval/runner/harness_runner.py,sha256=n3PiJIJxDRPleTaL2vvG8UGyLbn9t89wqt90XzeDlfI,10232
maxtext/eval/runner/run.py,sha256=_ZHQSjfPhtXdZJbFqfL9_boYcWPF5SGXenU-91t43Ic,3449
maxtext/eval/runner/server_manager.py,sha256=K5eUZarklc4jecBtBIko2tXEojHztB51C40cUjWr98o,26904
maxtext/eval/runner/simple_evals_runner.py,sha256=2k0e85jnqL1Id_TjTgr-GuERpaK8S2zz_Lu7OK12o-s,22128
maxtext/eval/runner/warmup.py,sha256=UMsRJUByl4ujcX7swloSFB7GTMbg1zlBQLtR8065vE8,5398
maxtext/eval/scoring/registry.py,sha256=D0jJ2RSHNXPVNqWqM-Fiyhw9va-v2ZJSEbKvS8BFpiI,1743
maxtext/eval/scoring/rouge_scorer.py,sha256=DKDH5ZNyy_UFlHi78Zfi-OV1Al-ePM5sqDoYvbNpSUs,2036
maxtext/eval/simple_evals_template/__init__.py,sha256=SkIzyAq-EeL1XQgLt6U-NRDVaew2QCEDFqMR54NjKaI,1929
maxtext/eval/simple_evals_template/aime_eval.py,sha256=oY_-FVI3g6wkGsps50TmcAuyIKnODV8ao_20VdmdXLc,5499
maxtext/eval/simple_evals_template/common.py,sha256=KPha5kbpZaKFC-2QxVbXqE-ytIeKQcCH7fCR-bxPwSk,11570
maxtext/eval/simple_evals_template/drop_eval.py,sha256=3_pt1H7Ed6zfEjdJhV1WKWl1aqrklkJX5eogrujYt0I,11508
maxtext/eval/simple_evals_template/gpqa_eval.py,sha256=S6ekQ21T9koo9sbtNDUOIS7jp4L_D0TimdVX79TDsRQ,3961
maxtext/eval/simple_evals_template/gsm8k_eval.py,sha256=-qxaFns1ecq3gsyCyzY_cleFU9MdOkD-mfood5gGc7M,5196
maxtext/eval/simple_evals_template/mgsm_eval.py,sha256=6ZAtU937Y5wCmRsaC61srhnjPmkUikuYk5Hj_UMi5sM,10677
maxtext/eval/simple_evals_template/mmlu_eval.py,sha256=2Azm0Tg4a3mSnA-aFmBe7Q9HGo-moQjhPkssNJdk2ks,5834
maxtext/eval/simple_evals_template/types.py,sha256=7Xw8E1pO21N0ZXZfrB-1vT2vdMKjLd-rfVLhQtBzSE0,2144
maxtext/examples/demo_decoding.ipynb,sha256=VxQnCibw2U9l0F9JydrTbFTyNdrxKCeOJ_X2TOYdIwE,13519
maxtext/examples/dpo_qwen3_demo.ipynb,sha256=hBNcsDja9npzUUOGdFuuqRAT-hj_pc-Wg68cowhOva8,563118
maxtext/examples/dpo_qwen3_demo_no_eval.ipynb,sha256=yVlzd6D_jRHtQsQPR7diDxXzxXImObfs02YQAcg99pI,6683
maxtext/examples/lora_llama3_demo.ipynb,sha256=lvlwdtMwol5C0LZmC2PjNQqedvv9EQUz-5ElpuTOpOQ,21545
maxtext/examples/maxtext_with_gepa.ipynb,sha256=NFqFQP8x95AOT3OkfATNTheqblBxCOqOLCnX3LYDhy0,24374
maxtext/examples/native_lora_demo.ipynb,sha256=e3cpzjIGz6eulk2iQgl-omuYqfFM0IQs8C2CQmiYeOs,11268
maxtext/examples/non_spmd.py,sha256=to9A6lW1aOfdJlQna2IIkVQ4FnrfD5FxYLm7JqlSfP0,1832
maxtext/examples/rl_llama3_demo.ipynb,sha256=8nEmpONuwAuLf1LZBYSbPXHPkCaFq2A4DV3vzFUA8QU,12365
maxtext/examples/sft_llama3_demo_gpu.ipynb,sha256=_Y0g3fsl5DODxzYvKXHP2HTfPjIVCZkV60S_4A5xdOs,25438
maxtext/examples/sft_llama3_demo_tpu.ipynb,sha256=HM2xC0sL0szOboK3MVu1BOau1pV-ClrfedU-iCSirsA,12820
maxtext/examples/sft_multimodal_gemma3_demo.ipynb,sha256=gIr1EVosmBD4k7RR2UZqgcgrUFJNpKgD4_1AolOW8CI,12143
maxtext/examples/sft_qwen3_demo.ipynb,sha256=Gmx4fElc0hRZ2fIM-ftGvKQxYqD82EW2FZywZdlgWmM,17710
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/gemma-3-27b-chat_template.json,sha256=_ha69yjbSUV83jKALNfvwKyKeph32-Iv4zIrLZ3GzNk,1615
maxtext/examples/chat_templates/gpt_oss_rl.json,sha256=IrqqEmhFaKKRVdEO0ekNeVeF9V3AAlOTmgE_DQw_05A,772
maxtext/examples/chat_templates/gsm8k_rl.json,sha256=3qynSoTQjVdlfDoVgmWwCKi5N5PZPTmw8eK-4I2Yg2A,394
maxtext/examples/chat_templates/math_qa.json,sha256=AW87IT8PSe5Vf2tIWUz6JQhsO41MkrOEDZJoDVBrMXw,503
maxtext/examples/chat_templates/openmathinstruct2_rl.json,sha256=ErM2fXYLniTbHwJAiECXIR1Op_JbQJ0uKuEEnquylxE,338
maxtext/experimental/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/experimental/agent/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/experimental/agent/ckpt_conversion_agent/README.md,sha256=rXAaJI7PtVmRokrMoLhzUE0tBnnIucmC3Dx8UcN__LA,6761
maxtext/experimental/agent/ckpt_conversion_agent/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/experimental/agent/ckpt_conversion_agent/analysis.py,sha256=CI3-z65d_j3YYyIKCOqBjojirgi3ATXrWtpWrv-g7T0,3245
maxtext/experimental/agent/ckpt_conversion_agent/base.py,sha256=Eajfpt6mRcSwLzZdl1HNe6aiCBqfTL73oyYLeQR25yQ,1779
maxtext/experimental/agent/ckpt_conversion_agent/dsl.py,sha256=aoBlCW_Rn_xm84I7wrM2At_RTmiGgX_5erX-yCTs96c,3246
maxtext/experimental/agent/ckpt_conversion_agent/evaluation.py,sha256=PgSdJaFU3QVIMjxWrWKJjU3rwY_sD8g_QtNm73sJM84,3758
maxtext/experimental/agent/ckpt_conversion_agent/mapping.py,sha256=xjgBxW-LVJqnbAPRML2kWgJ9c4qO5wXi6zZV_RVdU0s,6697
maxtext/experimental/agent/ckpt_conversion_agent/plan.py,sha256=NBo9duHXMnlo_fE67HbVr-GePwLT0wLBSU4srG_soP8,4199
maxtext/experimental/agent/ckpt_conversion_agent/prompt_chain.py,sha256=gTjr6zVS_H03ZKvxnJLKn2yE8xgxr-PaPFfpoorSOEA,7085
maxtext/experimental/agent/ckpt_conversion_agent/step1.py,sha256=pTRZIKjoa5gAWqI0soOLHC3ISor7oVk1MBoB4vG6Yn0,1655
maxtext/experimental/agent/ckpt_conversion_agent/step2.py,sha256=yDVt5VS_UC-dMvKfL-xAfMn4xt_cQcGcLPKQG69-ro8,1924
maxtext/experimental/agent/ckpt_conversion_agent/transformation.py,sha256=xhYS3LCm0mdsYpv_F-isFEdLggVlyKdyz2zHIEtab_c,3655
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=NFUS75vUdUuhfYi9-iJVUIkOTJ7KjpXwLc1ZTkq5Qus,2768
maxtext/experimental/agent/ckpt_conversion_agent/baselines/context/param_mapping.py,sha256=ziSq89Gk--nUhpQ6ki9JKLJ7ene4fHpcUvC01EyS8uk,17071
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/omni_poc/omni-gemma3-qwen3.yml,sha256=hFIo568V6DoRPJ-Ii_YpTkrnmv9DpzazGfd2qqfW3Tw,1219
maxtext/experimental/omni_poc/prepare_checkpoint.sh,sha256=V7amE1VcEMs3tUADHaaLjaFgL3MZQ8vOBRs-DJYgBQM,4578
maxtext/experimental/omni_poc/tests/custom_vision_projector_test.py,sha256=_GfSWlqp4qQ_WFQ8LUjtvd0BBmNZiUYEB-piSB6LPs8,6424
maxtext/experimental/omni_poc/tests/stitch_checkpoint_test.py,sha256=8sxShS5Ko7_B8sKr0UFprm4usgXr0e9kxfWPQe4rpIE,17003
maxtext/experimental/omni_poc/utils/stitch_checkpoint.py,sha256=hCZze-qnl0oAT1X3ug09spcAO_JEoZqQ8iF7CaKVNs8,12032
maxtext/experimental/rl/README.md,sha256=gYKEIqGx97U8f9IQdfqKm3MOSpjyC8lk24Y3qoZ8Z4A,3869
maxtext/experimental/rl/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/experimental/rl/grpo.yml,sha256=KBAein8FPWctNkPU2l0kmh0rYS2WPv9hvb_L-iTCqz0,1033
maxtext/experimental/rl/grpo_inference.yml,sha256=osU7PKzJZ8D3vyIzE7o05ZE2re_n54Q-PPW_-eLhsE8,1021
maxtext/experimental/rl/grpo_input_pipeline.py,sha256=_4PTckhBh_vghvdKfFIR-X-bvfoyiAGk1XnGw7iZI9c,9368
maxtext/experimental/rl/grpo_trainer.py,sha256=kr32e5g3byfJkNnqu5HSZBP_YdMy82WXbR26mS5kLOw,55459
maxtext/experimental/rl/grpo_trainer_test.yml,sha256=kP8bys1T2ZEw4Z5F31twZgvMnxURp5kQZmmcB9Z2mZM,318
maxtext/experimental/rl/grpo_utils.py,sha256=zglbG5_WCF0PC06RNZIhsCKYpOrwAzSZIPPBwxDtK3w,27519
maxtext/inference/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/inference/decode.py,sha256=exYt5c1GibDgMZSACQyt99OKGQvAU5cWXdQFpBFK76U,10792
maxtext/inference/decode_multi.py,sha256=nBsNKNyWp17OxksBxRW1dqffxj4KxwGTrr_CrQybOzM,8814
maxtext/inference/inference_microbenchmark.py,sha256=h7jGZxbgzCs35vL6LOuwQ_GIWJ3TvuQ1nHmYULJcFZg,18009
maxtext/inference/inference_microbenchmark_sweep.py,sha256=HWHYC_tkUiw5BWPDxNJ-hLONQtcXw1onv7kXh6Sa3G4,6415
maxtext/inference/inference_utils.py,sha256=ps5C4YcbAI1cDoy1Qcprf0hOrDGDqOhrVzhZEyLSd68,9874
maxtext/inference/kvcache.py,sha256=G3WOvSNYO4foZzIsBJTIZ0jRmza38jKmJz8X0Mkoab4,54029
maxtext/inference/offline_engine.py,sha256=3hO2J3WvaYPPAxwdUZEu_EXjY7uXNHbsSIspJ3j8MxA,32355
maxtext/inference/vllm_decode.py,sha256=wgtGIET22pU0hImWlwsqpRNj2CgwgO6d8ttV2KdjYx4,12629
maxtext/inference/gpu/README.md,sha256=JnQyfxEyZ2l8aX4PME6O8gXb2Fd1dJq66kbZFJqvASU,110
maxtext/inference/gpu/microbenchmark_llama2-70b_h100-8.sh,sha256=953_YluIXZevplZt-8zhMuhoTEXLycF8_RjQdfAd1F8,4335
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=A3CSf8ax6TxNwpZnDkllZBl_RJCuDqQrJCUPzMsu-i0,92288
maxtext/inference/maxengine/maxengine_config.py,sha256=mho_vg7Ou-YrtmfuaYxkDsM1ko98BGAWkaZ4iyQtcdo,2935
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=Sxqqse-e8nlkY39F6LulSS_UghSNeorYIF-ZfuOSDpI,7547
maxtext/inference/mlperf/evaluate-accuracy.py,sha256=nZJP5qgLACSUYXcR8dU3pkQEG2U_ONQUhPMezhLcfN8,4447
maxtext/inference/mlperf/llama_offline_run.sh,sha256=d8S1vpvbIt9Pb1qTqVPAg_rwOL6cSe3QbjNzWoxcQHA,6233
maxtext/inference/mlperf/mixtral_offline_run.sh,sha256=0xSF9bhPS6EwQCEyCIipF_6jdAHEU1qCboDS7BcFYJo,5913
maxtext/inference/mlperf/offline_inference.py,sha256=D3wiB3j_QmOX15qRttQ5HoyVtmpZkHDbDHGPQB4JoNw,13590
maxtext/inference/mlperf/offline_mode.py,sha256=EFk_bFtprJBuhcHQLXEwwUzCEuA5wkfNHv53tz30TVM,18003
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=Fd8sW-HQbMV8JfcWKwQOSerLB0eftO0MUhANvc4IDfQ,1538
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=IQUBUeLzwZHM4_bRm11WJRjO8u9CcYuSyGLz1TH8pW4,6729
maxtext/input_pipeline/distillation_data_processing.py,sha256=m5g2Y7qtAzQmte7zEIFNA8-1qR5iSiRQ4pPfQ98-_lc,5419
maxtext/input_pipeline/dpo_utils.py,sha256=YPzqlrVtksmcPOeT1w7AQHgAAuKbyhOS9eDHkAjK9yE,4132
maxtext/input_pipeline/grain_data_processing.py,sha256=2vj4sg9BAQ_nCKP5HPxNUVnt3dTCiSOFSD1P2LSRPZ8,24438
maxtext/input_pipeline/grain_tokenizer.py,sha256=Khe6_y8aV67YUvEvKcPiZoHsakEulKYRCdI76Xw3M8A,4426
maxtext/input_pipeline/hf_data_processing.py,sha256=6goHS5dwIeOJi-ghbxtR3e_orXjL_cwI_jWrRY1rB-0,21654
maxtext/input_pipeline/input_pipeline_interface.py,sha256=SzOIZ4pjNgNRT86UcihN6CpX1PH_UePz_LtC5LAZgTA,5922
maxtext/input_pipeline/input_pipeline_utils.py,sha256=l7zjBIl47FJDRq_T4rmOLpxyFjldAtRsYYoQEYQBf60,46972
maxtext/input_pipeline/instruction_data_processing.py,sha256=bc8lgdOFeceAZKk74DkSxDBRfS96PHXrVQJN0tx_WGs,7550
maxtext/input_pipeline/multihost_dataloading.py,sha256=bSHy8W7u3EoqTrgh_qln0YOos80kG83_ktWJ9HgYRvc,12147
maxtext/input_pipeline/olmo_data.py,sha256=OLe_HIzZvZPkPVrHekV1M3N0FxNntfROTXqSQDOOTnE,13917
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=_QSht6FUUavf77nxNRqVQcA7W7WwsH3PCV62X1zlCeU,5927
maxtext/input_pipeline/tfds_data_processing.py,sha256=3JHhr5LMMzcKKJK1kaxPrdfAcsbS3abvhNR6GEWIt48,12344
maxtext/input_pipeline/tfds_data_processing_c4_mlperf.py,sha256=wCr6G8RmPI0lytb6KU1dY0ykptNJxt3osWOKrON_VNU,15261
maxtext/input_pipeline/tokenizer.py,sha256=AVl0TTBE_o_r7FM-NkP8hHJvQek1MU4e5cogOmQxyv4,9204
maxtext/input_pipeline/packing/__init__.py,sha256=otX-EDUVsL8kHS275NFH8VnNE-7RDyQmKir2uN1UyGg,581
maxtext/input_pipeline/packing/prefill_packing.py,sha256=lw5Ayitkn3PtuVF596BozvB98TsIbeMiiTshNamQwNA,14547
maxtext/input_pipeline/packing/sequence_packing.py,sha256=ktvrcaZSgzNDrrFPsltnN8DpakrH2Etau55bOFSyPbQ,8003
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/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/integration/tunix/tunix_adapter.py,sha256=nIsCXiRrQ9k7HozdHSdkYeIEKnUttRlXiF3UV9U1sXk,5382
maxtext/integration/tunix/utils.py,sha256=5kDh5JfmjbsFpoeBF41EBhH4I_qNjZ5tNUbrtrPXr8c,11359
maxtext/integration/tunix/weight_mapping/__init__.py,sha256=esXyDl9AOsIkPwqt6-2_ndiHpiqodUm61iseemYf9UE,2136
maxtext/integration/tunix/weight_mapping/deepseek3.py,sha256=3VgCC68weuC-lj1PlksNnuvAtZU6nHgKc91mf9ntbe0,6869
maxtext/integration/tunix/weight_mapping/gemma3.py,sha256=nrf91Y9IMgAtdlhjYFivyQg1gzNXwCSB6B32KjM1nU8,4498
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/convert_utils.py,sha256=YamIJLvWaE-anyIlSbKVB7RQ3_wvOGQaz01FT_YsWk8,23302
maxtext/integration/vllm/hybrid_cache_utils.py,sha256=I2OALThHNBwawn29q-l142sv013uGBsgujWnrCnjVXs,2214
maxtext/integration/vllm/maxtext_vllm_rollout.py,sha256=jDmiB6Oe0oAxhYz6bXypXAawyqoVSYwLmKOEsZMnz90,30860
maxtext/integration/vllm/setup.py,sha256=elb-FqVRJGL9QXT_vqrk2i_FSyYk-MSYXGmTa1mXf9I,910
maxtext/integration/vllm/validate_converter.py,sha256=znHL20KtlW-HDFcakbMv9wvL0wKD15DJVaqj_IiyW-I,47759
maxtext/integration/vllm/weight_converter.py,sha256=wvVpboIq8GqTJ6SjE3VRwN6hi0kMBSVfFJ860oHcjxg,40419
maxtext/integration/vllm/maxtext_vllm_adapter/__init__.py,sha256=Vj-OIsnQF3eJnX059Br2AVsEL17e3ScfOON4vigVJI8,1435
maxtext/integration/vllm/maxtext_vllm_adapter/adapter.py,sha256=GbMhgkBonNPpDqOEaFjf_SHB1uzjAQUtTictdNtRlno,20377
maxtext/integration/vllm/torchax_converter/__init__.py,sha256=otX-EDUVsL8kHS275NFH8VnNE-7RDyQmKir2uN1UyGg,581
maxtext/integration/vllm/torchax_converter/base.py,sha256=Qt7xM9L-t7GrypHNm139mVzBZp-vUTfi0ojuXbGQP0U,2015
maxtext/integration/vllm/torchax_converter/gemma4_moe.py,sha256=Zs519j8rYP6fg18ATx0eqa4en_Nb7QPHjzT87lBhiFQ,17684
maxtext/integration/vllm/torchax_converter/qwen35_moe.py,sha256=dw-Y4_FTmsSKiu93gozy8K8TqNyDhcL48_e2_qEFHnA,12049
maxtext/integration/vllm/torchax_converter/qwen3_moe.py,sha256=ykpCRiKSFyHQzkWmZk5VJYpxBRb98j9I122Wd2Tn4fw,9162
maxtext/kernels/__init__.py,sha256=otX-EDUVsL8kHS275NFH8VnNE-7RDyQmKir2uN1UyGg,581
maxtext/kernels/gather_reduce_pallas.py,sha256=bZ92Z5EiPZVq5Ftn1ErG3r3Y3b8HAVDTotyc9yt-wcA,7797
maxtext/kernels/gather_reduce_sc.py,sha256=4Ywk9b_qzU7YYS1LQGRzlgBSaykvFnB8jWL3c488We4,57003
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=cxrYzzsY3miFAAw8kMuwPRYVHb6GpaqpxLHH2sY-Kt8,10118
maxtext/kernels/attention/ragged_attention.py,sha256=IzCShpFJgHKLdv5uDXrYmvpeWWfgYy5IcbBKCwfA6TY,14138
maxtext/kernels/attention/splash_attention_kernel.py,sha256=SI80cdVspQAiqh16TKgIS7IGs6rDrBM0HXwHZROPYlY,80331
maxtext/kernels/attention/tokamax_ring_attention.py,sha256=PxXOTTp6SlZ9JspQImJjHBrkotHgTemCcJzAGN2Pckc,14315
maxtext/kernels/attention/ulysses_attention.py,sha256=xm-ms2-elBQn56W3kxUsqGp3XRD9dbgjTBhPZED-GvY,8006
maxtext/kernels/attention/usp_attention.py,sha256=wbvBNHX2PlMDAQ9ezE2tV4iRA1yCNhdJ-u66_KZyD2I,6158
maxtext/kernels/megablox/__init__.py,sha256=nxu9b3_E1SxAK4kUpf_rFUloFeNcDGXJ8CYlrGbD75Y,649
maxtext/kernels/megablox/backend.py,sha256=lCIvVUqfaqAdUNPxe2cJW7j7o-Wum16dZ1Vc2P9Tk64,29333
maxtext/kernels/megablox/common.py,sha256=7ofB9dWKhxInTNCZMIIb6CAKU2t1A1GAUGMytuunruE,2044
maxtext/kernels/megablox/ops.py,sha256=cUcjG9kXw9zLXg0OsXt5ppiqB6-OJ32aYqc20RB84VI,29927
maxtext/kernels/megablox/pallas_mosaic_tpu_v2_gmm_kernel.py,sha256=a3PlMA3R0htkPitVf96ine3Yj7smFTGngtYndQv30Zs,53521
maxtext/kernels/megablox/pallas_mosaic_tpu_v2_tgmm_kernel.py,sha256=rsgcw2oEK4NbIXwOnk_Xs_E4uzihvkXKbIFAyHFBUTM,29924
maxtext/kernels/mhc/__init__.py,sha256=A0OqSKmAPGNi5O6mD_GPWtK4qyon-TFQqbiPIQvUzss,1321
maxtext/kernels/mhc/api.py,sha256=peHNNSWYgFS4SEQmT-K8NNHZmom8ha3s-Lzwul1wtqc,3870
maxtext/kernels/mhc/common.py,sha256=GPct8uCeZBZY4GsewOK6ZmStciABjQPMqoJhVMySQIU,20875
maxtext/kernels/mhc/mhc_kernels_bwd.py,sha256=zrropRgkKVaAK9W3DPqePk3i0P8-hB6ECmRg3GUW7Go,15458
maxtext/kernels/mhc/mhc_kernels_fwd.py,sha256=uxTe6j4ZpL5IPW0Tz3WY5waEvpU2Mzf_4Dg9R_FgKlg,10970
maxtext/kernels/ragged/ragged_gather.py,sha256=W0U1-295IyVd-lahKq9RRKccUAwa1lIAeiIlqfuANq0,18323
maxtext/kernels/ragged/ragged_gather_reduce.py,sha256=az74rg92cCLXMRfJRd-6V24kQ7asgcc9GBkbUK5JXvQ,23710
maxtext/kernels/ragged/ragged_gather_reduce_v2.py,sha256=Ij3xDQjvkbH_IZ02j48XGLTXlJT8_7VbeWomMFQtRZc,30153
maxtext/kernels/ragged/ragged_sort.py,sha256=qmb_ibpHSpyMlKzwilYB069MEhMMSSplKXXz5mcUYE0,25212
maxtext/kernels/tokamax_splash_attention/__init__.py,sha256=kbXwi_5D5swIvFvoI7C9s6vkcH3frHiRc3ph9WygepI,751
maxtext/kernels/tokamax_splash_attention/base.py,sha256=qg58fWW0xBYfNYR51DH7TDnDuqDFZp1Fwe4xjSGXJ5w,9027
maxtext/kernels/tokamax_splash_attention/ring_attention_kernel.py,sha256=HUK8M1Eavi_N1ePix56qoq1WzRLlVNk8dUigTEs2l7o,32577
maxtext/kernels/tokamax_splash_attention/ring_attention_utils.py,sha256=Ie7TVFIHtHBHfAdecLdGbYwVkbvQlT-7ye_POjKgiyM,3792
maxtext/kernels/tokamax_splash_attention/splash_attention_kernel.py,sha256=sK6_ffspy_7Iq76EoBCJvaGnhvQDpTc7RS4pwC63O_4,72840
maxtext/kernels/tokamax_splash_attention/splash_attention_mask.py,sha256=NJ8bDgSEl6GYZOy0Z9MyYToMkC8N7ZrBlNHcEYfBLjA,15514
maxtext/kernels/tokamax_splash_attention/splash_attention_mask_info.py,sha256=qj1XzwNqMulIhDwSuc68tuwTenBEaLxPXdneU0s_ffQ,21951
maxtext/layers/__init__.py,sha256=otX-EDUVsL8kHS275NFH8VnNE-7RDyQmKir2uN1UyGg,581
maxtext/layers/attention_compressed.py,sha256=U20CssRJ8UYEp2VWBFHQYuAas_jKiSlpOhJvnI7A1dc,68488
maxtext/layers/attention_mla.py,sha256=j1PWfeXM-430XIgHoNM6RQl45WDX-mYljZ2TGle1JUM,55760
maxtext/layers/attention_op.py,sha256=TGK2mIRf-O466reX4EAD6b2Ic-agVU9j-Xl7i27CmTo,123015
maxtext/layers/attentions.py,sha256=maIaivkrrB-pppgcfFAGYOXF0l1Qamgjg2KWAHhIgAs,58057
maxtext/layers/decoders.py,sha256=qQZIhMEO2l9eUcJwkTokyVqls9-APKJX1-O1hvKy9Uk,73438
maxtext/layers/embeddings.py,sha256=bdvNAz0b749-OA4CB_KkXA4WL1M7vn_YcBTK5LAtnaE,79591
maxtext/layers/encoders.py,sha256=UPiV4oCvN-CIvhpYTQP1ixeNqBeyE1ucqNzyFT9ZHC8,11549
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=_YDKjW-LLPbNgfdyS6sD2ECeQzmM-UxNVdJ9lsreVuw,17609
maxtext/layers/linears.py,sha256=UuBQxWKyxFvDdSEWotxwoa1l0Y90Akw3VfaezGufkwg,31026
maxtext/layers/mhc.py,sha256=hgdq-_b5xeSC8mwN7HpNbShMcRXTatlxNUfxQKc80Ig,15138
maxtext/layers/moe.py,sha256=qqDO34uY_OISHMAvEFxJM2dqbRG775_gzs8ZWC-eY4s,141262
maxtext/layers/multi_token_prediction.py,sha256=v1D3cJP0gH4g9I0gsFj9ZH1-QmtfYxU-3Lw34E5X35E,21967
maxtext/layers/nnx_decoders.py,sha256=MiFMpTiOra4A5jSXqvHLaECa9FRUMrFyr0QRjzlui2k,91185
maxtext/layers/nnx_scan.py,sha256=55Tn1VRaYdqm_kNL7Yfv65pcGi-Th_f6GNRlXuwzXMw,6937
maxtext/layers/nnx_wrappers.py,sha256=9Kd-GIZsmr_Ego03h-G2UMzeOyqrXIGCc0hsW8ebFc0,27296
maxtext/layers/normalizations.py,sha256=KUyQy5jg-MksjUSwoiEsYrpVmbrU46zD8ODYnIy8Qak,8325
maxtext/layers/pipeline.py,sha256=Kfaw1jiQfYq4B-gbR1Hi6IK6zsoCp7jnT4_FNs9oVtM,84913
maxtext/layers/pipeline_deprecated.py,sha256=nHFLMd4HpYi3AW6xDK5liblfKLJoQ20cp9JzmBgQau8,44243
maxtext/layers/quantizations.py,sha256=9u_PeSNkhuDuruzxtg3mV52SuNdm7H4ltoy9WqGrLpY,44322
maxtext/models/__init__.py,sha256=otX-EDUVsL8kHS275NFH8VnNE-7RDyQmKir2uN1UyGg,581
maxtext/models/deepseek.py,sha256=Z_8-SswcsLgUwAsr0KBArXCiE8gxnsbFPw-0srJJpa8,22820
maxtext/models/deepseek4.py,sha256=g977LVloZQDVQLgtC-wMmxSrKVcUndWHhzu2_XLoEO0,8943
maxtext/models/deepseek_batchsplit.py,sha256=x0wBi4wk35KnyZQcxTI0Bf7Lnc_wARY6MsHMDh_UgN8,95724
maxtext/models/deepseek_batchsplit_fp8.py,sha256=iD6Ojzl6xsyduMYZk4JIKNW3n-T3AWnNAB0wRwKZU2s,37404
maxtext/models/envy.py,sha256=zSZxFzPRRu_PcRuEjDO6o0v35QSqCfH533PBhxvDiZo,11859
maxtext/models/gemma.py,sha256=uprXgSuZinVhtEbKkXwkuRVT0f8NSOQokoS4Gn-uHYw,6142
maxtext/models/gemma2.py,sha256=PcgF8cGVu7Wv7PzDI_GVmhdGXbFQqJs-BZlVeULPuRY,10908
maxtext/models/gemma3.py,sha256=ZGGFl-eLQuSBymRO3rMckwFKxuPLClvEpn0pD1zYl00,23806
maxtext/models/gemma4.py,sha256=m36FS9qbzRP2kaHqLi6-BQ7R8lDK2vOsn3y2oT4bqc0,27018
maxtext/models/gemma4_small.py,sha256=h_4cu3auSp4AYvjk9KtCqVZ8SXOhPZ8yS76PWyC2cMA,16587
maxtext/models/gemma4_vision.py,sha256=Lp_waeVUr6kRvwkkteAkAILNhNHDrCQOl0LQre0rgIs,22183
maxtext/models/gpt3.py,sha256=0vERn8JwfFZiIifr4vfXRD12u1lO9zxyNv9o8UXS2TY,19003
maxtext/models/gpt_oss.py,sha256=3bqG6TL5CV8K6q4YbxEaCIllpfoMrjlyoJjwthXZjak,10004
maxtext/models/llama2.py,sha256=5-T0Vu_2x5Pc7Y-UEx9Ic6Mtr4OGVHSDHMtl57dDN7Y,8504
maxtext/models/llama4.py,sha256=70GJVQUGJI2TCB0UD-NI5pNYqMR6J5UCfhSfEPhw5RY,30890
maxtext/models/mistral.py,sha256=KaYufWg9XqvguuZQq7MhtWs1XVK6MyKZl0DLIDpFSck,7236
maxtext/models/mixtral.py,sha256=ztw5rjP0NtGef0RIuU0y7KrfRkUt1-gjiGqQnq2WFRY,7114
maxtext/models/models.py,sha256=wkbJDUxM73hsw3H8Es6_rX_eXp5C8KJ803q-xj-uOyk,27308
maxtext/models/olmo3.py,sha256=qneKcZH2JJkMHCikT8i5V6g6BsXFvCSFgMU9IoBtDlg,10447
maxtext/models/qwen2.py,sha256=1Zbep7N07elYV6PoC-jOgaY-lNAZu2QZk3nb70FUHSc,7520
maxtext/models/qwen3.py,sha256=zDejFXk2WdgTa382oTe6jvVZjlf3aWhfjeV86RDL5yU,97346
maxtext/models/qwen3_5.py,sha256=q9Qk1YRig-EhnfBsbPbZQatyOVAXb_pquxlAPJWBBF4,8549
maxtext/models/qwen3_5_vision.py,sha256=VNoELQBiJB-cND57Dp3rLuqmdS_hBpEUK2xLnICbshM,1448
maxtext/models/qwen3_custom.py,sha256=PXcPOYdGvcZvh0PrbDwaYu6PtPZdpsMC9lFIeJjoExc,9714
maxtext/models/qwen3_vl_vision.py,sha256=EkjSiRQpI9cVtAORuXB48mlgXI0s2yjTcPm0So50ktU,1202
maxtext/models/simple_layer.py,sha256=Oi-zw4fiXxuJYzugGOMOIUa5-mWEgGPjuyfGN1uZ9Tw,4905
maxtext/multimodal/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/multimodal/processor.py,sha256=X1hW1G5K_ZEyoZR9BrXDdQ-Roc92A5h1XA_gOitPLvU,15229
maxtext/multimodal/processor_gemma3.py,sha256=suFX1VPd_0TdqFmt4gmY2Bq4ErBQWlUahzxTjG25pbs,9808
maxtext/multimodal/processor_gemma4.py,sha256=SbBesdMQaqys1hsckMENK-hYS4D6BZnMO3AdXO8RMXM,6523
maxtext/multimodal/processor_llama4.py,sha256=Mbm-kW_UzqUmvARxf0z1w_TzVLWbeFwD8hEls7UgE7A,20705
maxtext/multimodal/processor_qwen3_omni.py,sha256=4PQWKqgSWR_Y61ZR5uZm_Ks1e3jAcrSTpR2Ne82DUZc,55038
maxtext/multimodal/utils.py,sha256=OkWjEZpDSDeXzwGx6OOW0F6NBncBg3j-gHRXTecgxD8,34790
maxtext/optimizers/optimizers.py,sha256=COtH4K3i99WJHriZEr2QgAvZmGZIH_Hk5DfOFE1z0UA,14352
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=Htb7C22eT9o0bmZj6AMTcDTLXZWB-ZXmTSahSH5hw8E,6255
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=68bYjqpmNaNBzR_CQc6iq2YX0Ni05uaCt15ero7RS9k,11365
maxtext/trainers/diloco/scripts/run_olmo_qwen3_30b_streaming_diloco.sh,sha256=W6-pEcbZq9gedTKnX06Z9tsjF7fo5OTpfTYcszVBIGQ,11253
maxtext/trainers/diloco/scripts/run_spmd_streaming_diloco.sh,sha256=eVcVnh6W_AWKerC0NtnOgUyJPHOLjzhJXa-FBvhy3IQ,9173
maxtext/trainers/diloco/scripts/run_spmd_streaming_dlco_moe_c4.sh,sha256=_0fC7uAlwRNLtBgHwuubLIogL6t4SicCYMB0FJX-PUM,7731
maxtext/trainers/diloco/utils/__init__.py,sha256=2IK4ccQkbLO4plGzuETJc5TbiIjRoUoqYnufSPWIBqw,2110
maxtext/trainers/diloco/utils/fragmenter.py,sha256=CsmgdHKaF7R8_F__PhYoa-AmOda8WFhPH3gtiyyWUe4,11605
maxtext/trainers/diloco/utils/nnx_state_utils.py,sha256=VpLSVEDbuIUjVdMGWGi7uzK9_JNf42WqO5sTkX6F83k,3662
maxtext/trainers/diloco/utils/spmd_diloco_checkpointing.py,sha256=6GPB-f-p99zRxGMFD473hXzbxgVw7nvbMxSz5JXEmuM,10500
maxtext/trainers/diloco/utils/spmd_diloco_sync.py,sha256=xRzCr8x0008FFDpc2ahevLDkvWoOwzOlA9teGc7EsGk,7127
maxtext/trainers/post_train/hooks.py,sha256=Hegn3ZN0u_8AWzkEr4cDZvm9VmXYCnHUrfeP8W85hnY,9454
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=YdqI9g6_tI3gC1IFO8UFe47QGyFrzUCHztaPj38iM0E,31912
maxtext/trainers/post_train/distillation/lti_utils.py,sha256=XRk3bpL_U0BSJW1n6TGWvJRkn7KDw0ABTASXCo8VUk4,5692
maxtext/trainers/post_train/distillation/save_top_k_teacher_logits.py,sha256=-TA2fkemguTP1Y7d6ij23G_NphMElcNdLtSNeJbb5mE,12575
maxtext/trainers/post_train/distillation/train_distill.py,sha256=FYcEFWQEkdosnxUL3zjN_931-zp9BvDLz8TlHQKtYpc,39499
maxtext/trainers/post_train/distillation/verify_saved_logits.py,sha256=-WxjZhK9dvaQi8QD48cM14VqMkkgQQG--ZPZWV3Gw8I,4541
maxtext/trainers/post_train/distillation/scripts/run_distill_xpk.sh,sha256=532p5R1WDC5xnyvPTyf1nSLE6ckR4yc-NA7ggJGRz0Y,21650
maxtext/trainers/post_train/dpo/hooks.py,sha256=uVQvOaL50kUnl5-yaUYxp7zMtERQG0f7Hr9EC5LLFtg,1118
maxtext/trainers/post_train/dpo/train_dpo.py,sha256=WfxsWWHHj4pY3q3JXYxtGpOfl43qylZkX6jj0A26Viw,8433
maxtext/trainers/post_train/rl/__init__.py,sha256=cdqqYdkFUc32oyFDQAoPRtzLJooOrHLWRUL9aVNpGM4,581
maxtext/trainers/post_train/rl/evaluate_rl.py,sha256=RTXjuQwpZKRGaX84SiL9Rl9St3kB5UEB19dOrhpFuuE,12568
maxtext/trainers/post_train/rl/hooks.py,sha256=nUP3fwnlXr3Zh3g9IwolyLHCo-QQZQsoNh6c1y53xuA,3972
maxtext/trainers/post_train/rl/math_verify_pool.py,sha256=aCElR0PNtxDLD6hjk-uPKK5LSP-gmBrz7roxvJrZz5g,10509
maxtext/trainers/post_train/rl/train_rl.py,sha256=B4llZeoRc2VO-6gVLVkBEPMg60plM3kKjT7ne2jbWY0,35744
maxtext/trainers/post_train/rl/utils_rl.py,sha256=WczUh47FbkqxgQf_foHkm4N3OgMl97aK401PxUL_0V8,31991
maxtext/trainers/post_train/rl/scripts/run_gemma4_e4b_rl.sh,sha256=USv82uxBL0jWOjdv59Yj1yU173oWY-487XOkM9qy6Xo,6985
maxtext/trainers/post_train/rl/scripts/run_gptoss_20b_rl.sh,sha256=l8tYZ7THEEm8Oc2AZ72UQPz6DtOkAjyfro4vextfEPs,5136
maxtext/trainers/post_train/rl/scripts/run_qwen3_30b_rl.sh,sha256=TNY6DuyulQnBW4K0AW0Dl3Va7uZokoxcdtv7QeBVCqA,5880
maxtext/trainers/post_train/sft/__init__.py,sha256=QF_BPHJWccoZeSaarVzn0MStInlGsQojtUuScrvHPcE,579
maxtext/trainers/post_train/sft/hooks.py,sha256=D-eUISyo4hTTRiWRRflJYmWkhMhmbTC8Pj79RIpfOKk,1101
maxtext/trainers/post_train/sft/train_sft.py,sha256=BydCz4_L-rPx1aOK55y_n6H9fDadkW2Uy4W6JnTCBTo,14174
maxtext/trainers/post_train/sft/train_sft_native.py,sha256=vmSFJ_Jx3UtzcJEi6hAjIkyYvEiLitl-TtsW1ETwqD4,1257
maxtext/trainers/pre_train/__init__.py,sha256=QF_BPHJWccoZeSaarVzn0MStInlGsQojtUuScrvHPcE,579
maxtext/trainers/pre_train/train.py,sha256=wb5qFjqXvjk6xky2K72dDb00-QA6LTAk_MHL1BvJ16k,49124
maxtext/trainers/pre_train/train_compile.py,sha256=K2CtiFSYhSWc93AEXd_MvUW6ZDxJu53VYSfes7K3pt8,16236
maxtext/trainers/pre_train/scripts/olmo/README.md,sha256=VgP9FHD6vWtIQylWV3Z5l_Z-swieWmMgjDAP1BGPH90,8231
maxtext/trainers/pre_train/scripts/olmo/run_olmo3_7b_stage1.sh,sha256=ENUu-T5TJga4BhMhkd3mbBE780Y1VzllB2Wd6bkCTJM,11303
maxtext/trainers/pre_train/scripts/olmo/xpk_olmo3_7b_stage1.sh,sha256=VQ7KnXLVq8Gn4CRmIjV2S27AxUQZpNBAbNn4vJE348Y,13255
maxtext/trainers/tokenizer/train_tokenizer.py,sha256=owjtgBYWIsO4iSk13eOJoYMxTB-XbuO7jmhqO11KBTc,8692
maxtext/training_engine/abstract_engine.py,sha256=oXj9Ga1tY4WVQwRIzdOB6Dl-IDCByzscUkB0uo1jJpc,8500
maxtext/training_engine/checkpointing.py,sha256=EgerF4w3Hd7Td-nbym2LEkJy0rcE8lUN78DrfMM2irQ,8248
maxtext/training_engine/inflight_throttler.py,sha256=e9e7BXOrgEThQ5piAwmiQGT1CfReeMWF2bo_GynYips,2386
maxtext/training_engine/maxtext_engine.py,sha256=TzDmK4ommfHKGJ4ajrys3lzT_2sZXh-Y6RaJCaK_XJg,23275
maxtext/training_engine/metrics.py,sha256=a9EUCQVy5SA9yrMVxFbWUcyrMMTk3irDOyff6CbM9ZA,10047
maxtext/utils/accelerator_to_spec_map.py,sha256=tZMzaSzo6hHpQnWgZXo0xqS3WxqGl7xca-YRt4awkfc,26876
maxtext/utils/diloco_sharding.py,sha256=OvNCjmaoqrwXfwmiflxxitA0p9wFARpzfEhl5dQGX8g,4211
maxtext/utils/elastic_utils.py,sha256=ybQjYaA_kRk2jCrCyG4CuRvOcCM1vpapbdnPJCFKx8g,11413
maxtext/utils/estimator.py,sha256=lnFglxryX3KzgAKggW_b6U8jMnPThYnK06d-cT8xGIQ,20754
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=l95QxsmnagIXYysu3H5gDkKlSUJwYP8dL04vr_7Polw,21845
maxtext/utils/get_flops.py,sha256=HdDf-qPbKKNlc6Qso5_05FFmZCqxxmFy6zTHPpsOKQM,1366
maxtext/utils/globals.py,sha256=DYoI840YepeRejv5HjUiINF--bvDS3gprH-Ht13ULVU,4695
maxtext/utils/gradient_accumulation.py,sha256=VVAwSnO_p17EcAOtDYBMKSxFNa7TBq8hrRPep3LO-fw,9300
maxtext/utils/layerwise_quantization.py,sha256=aJTK7npZ7jirsUpDQ0VLpz2XnhO1Q9QOgtp_AznLgEs,19394
maxtext/utils/lora_utils.py,sha256=tmQgziDsCB0aqmjwzZvDpqlEEyhgcI25oZUj2dGhM-Q,40147
maxtext/utils/max_logging.py,sha256=xhab1Cm0xA-YRfnJEacWwNePEv-XxTXoC4nk9TyTqG0,1795
maxtext/utils/max_utils.py,sha256=dhGoFe2snf8YvkiNl3PdJvCqAgH3XsijmqNXHqlfOqM,47662
maxtext/utils/maxtext_utils.py,sha256=CmRtjPXdm8r3xuAjemVTqVTkObbeRy0k8mmTNeUMOsY,108484
maxtext/utils/maxtext_utils_nnx.py,sha256=Cz1cX45dsOpuatPgUrB_5vqKHV95DJR8Uompp5J6skk,9773
maxtext/utils/model_creation_utils.py,sha256=1EUW89JYQs8YDFGB3O1xayX5GfvdhTX_90IC1TnRsq8,56097
maxtext/utils/muon_utils.py,sha256=QQZFhUNahQHQ2UtHRYuK_oDJRe6m-ZTBZ6HvzSl9ddI,8890
maxtext/utils/pipeline_utils.py,sha256=_NeMip0qxc703X-SXYz8wwXdiM2VqRwLGjQd56MjCOk,21827
maxtext/utils/qk_clip_utils.py,sha256=6EQjTddBupux7UtDwwp205IpIkR6fZnWL3AW6L2hnHA,5996
maxtext/utils/rampup_batch.py,sha256=2ApGFe5DTyFF5jM2Wxgb2mZEsxOiGgtIyG2cLQPwtWI,4797
maxtext/utils/sharding.py,sha256=YUNax8VWu7fp9YNSL5REefJmYf_l5YVkB4PGXU64ydg,49035
maxtext/utils/standalone_checkpointer.py,sha256=OBXUkUgUPIbUcKZ1o7Hj_Jz5ZWFkr_-ChfR14Z2beRE,6725
maxtext/utils/standalone_dataloader.py,sha256=LPQaaV4hDO7Mu2DXNA8zYJC9tHQlgaCjBVz4orvcw7U,2746
maxtext/utils/train_utils.py,sha256=Bx3cZn0E7o3VcvGyAo0TZkm9StuUcVlxgGq1W4LUIZs,22550
maxtext/utils/vocabulary_tiling.py,sha256=5mNvab5VUqCv9qRmRBgOeHyK1SAp8Yg23gkUEGHhjgI,21486
maxtext-0.2.4.dist-info/METADATA,sha256=nHSG25qnvOcOr7eNDNpdX0GWU83I2kK5gjq9msa60Ss,74676
maxtext-0.2.4.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
maxtext-0.2.4.dist-info/entry_points.txt,sha256=dmttVcciWg1qO7xDg6MB_QrK-GiAIkouF6qglRa0RdI,513
maxtext-0.2.4.dist-info/licenses/AUTHORS,sha256=kbdMM4KiM3YreLkMyk3BtUmt_wQpBOdL8q5wEw_Fa_w,12
maxtext-0.2.4.dist-info/licenses/LICENSE,sha256=PWvDbyxiaVFtjMjuY07bL-EkG5UZKAr0_lvdglV5UkE,11347
maxtext-0.2.4.dist-info/licenses/LICENSE_HEADER,sha256=sUtr7zKNxe6y-XzxLKv_5YHhGL__UFNYIHhMuwMi3s0,568
maxtext-0.2.4.dist-info/RECORD,,
