kithara

View on PyPIReverse Dependencies (0)

0.0.8 kithara-0.0.8-py3-none-any.whl

Wheel Details

Project: kithara
Version: 0.0.8
Filename: kithara-0.0.8-py3-none-any.whl
Download: [link]
Size: 4871776
MD5: e6f6301e9681fb3946197aed4398bdf2
SHA256: 3fb0158c6681d9b56a30d2df42405c3fd4ef4c2d261f1a755fdeef3377daf695
Uploaded: 2025-02-19 18:00:22 +0000

dist-info

METADATA

Metadata-Version: 2.3
Name: kithara
Version: 0.0.8
Summary: LLM post-training library
Author: Kithara Authors
Project-Url: Documentation, https://kithara.readthedocs.io/en/latest/index.html
Project-Url: Homepage, https://github.com/wenxindongwork/keras-tuner-alpha
Project-Url: Repository, https://github.com/wenxindongwork/keras-tuner-alpha
Requires-Python: >= 3.11
Requires-Dist: flax (>=0.7.0)
Requires-Dist: datasets
Requires-Dist: huggingface-hub
Requires-Dist: keras (>=3.8.0)
Requires-Dist: transformers (>=4.45.1)
Requires-Dist: keras-hub (>=0.18.1)
Requires-Dist: google-api-python-client
Requires-Dist: google-auth-httplib2
Requires-Dist: google-auth-oauthlib
Requires-Dist: ray[default] (==2.40.0)
Requires-Dist: jax[cpu]
Requires-Dist: peft
Requires-Dist: hf_transfer
Requires-Dist: tabulate
Requires-Dist: cryptography
Requires-Dist: aqtp
Requires-Dist: grain-nightly
Requires-Dist: orbax-checkpoint (>=0.10.3)
Requires-Dist: google-cloud-logging
Requires-Dist: tensorboardx
Requires-Dist: ml-collections
Requires-Dist: tensorflow_datasets
Requires-Dist: sentencepiece
Requires-Dist: tiktoken
Requires-Dist: cloud-accelerator-diagnostics
Requires-Dist: cloud-tpu-diagnostics
Requires-Dist: ml-goodput-measurement
Requires-Dist: google-cloud-monitoring
Requires-Dist: jax[cpu]; extra == "cpu"
Requires-Dist: torch (==2.4.0); extra == "cpu"
Requires-Dist: twine; extra == "dev"
Requires-Dist: flit; extra == "dev"
Requires-Dist: sphinx (==7.1.2); extra == "dev"
Requires-Dist: sphinx-autobuild; extra == "dev"
Requires-Dist: sphinx-rtd-theme; extra == "dev"
Requires-Dist: jax[cuda]; extra == "gpu"
Requires-Dist: torch (==2.4.0); extra == "gpu"
Requires-Dist: jax[tpu]; extra == "tpu"
Requires-Dist: torch (==2.4.0+cpu); extra == "tpu"
Provides-Extra: cpu
Provides-Extra: dev
Provides-Extra: gpu
Provides-Extra: tpu
Description-Content-Type: text/markdown
[Description omitted; length: 3119 characters]

WHEEL

Wheel-Version: 1.0
Generator: flit 3.10.1
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
kithara/__init__.py sha256=n08A_hn2aIDfkFXVzAFg9yJEREW-bGUEwC9MB1X3krA 3102
kithara/callbacks/__init__.py sha256=i33tDlZDm_3kyaIrI34FHNxnaPEOKtbvtXfgaCtHZuI 85
kithara/callbacks/checkpointer.py sha256=o4ldf4l_ulxmbCraoaMJ7vnwH8JMf1YVk2PODuWhuC0 7348
kithara/callbacks/profiler.py sha256=mavYkjfajT0Shp-tBf_4Iw2TQ4tdFuPJKkRXzUgvNcs 6289
kithara/dataset/__init__.py sha256=2Fe1DX1Y56svp71QrEQkcZDaQy0jDB5LEduwoPGa3D8 159
kithara/dataset/dataloader.py sha256=ZoI3DyU_4h-IO_wZ3GrGcBYP8SZXX6-rrPXWKK_RYwQ 4076
kithara/dataset/dataset.py sha256=vnWLRQGJBuQiy8Udg-NtIHTnUGNbOIpYztLK-In_9VU 2475
kithara/dataset/packed_dataset.py sha256=UVwwsBIN-fiandy9_ojyaUyo51Eb7RtEboNJiY9UOUU 6143
kithara/dataset/sft.py sha256=u2DLQwm8wn53epc2EUhPoiWIV4pcSeb3ewGxyDgtx6o 4560
kithara/dataset/text_completion.py sha256=BiUPlfHhnWuIBu1p-z1yMsjDFmrilLLbq35QwaIcAYE 7820
kithara/dataset/utils.py sha256=NA8p1oqGkeH9DkkfkLs6ydaNolgPhXN5XOSe3vsqf2k 2725
kithara/distributed/__init__.py sha256=46kZIZofURb1cVTq1IOXMptbPlNOuEauaqrDMeto1MU 99
kithara/distributed/data/__init__.py sha256=3E1HAefsJW_YQKGoLDFFXNEgYf5Kz9hEAI95DK4rrmk 56
kithara/distributed/data/split.py sha256=SoTohBdAH585lktMdoTORAo4XuaqbcSkUPyx1LbelgU 696
kithara/distributed/sharding/__init__.py sha256=M1UhadimK_0Ux0QZ_Y2FiKkC86dTN2Gca8z0BFfD2Fo 95
kithara/distributed/sharding/_data_sharding.py sha256=H87zFpqGEN1py4Zc_otiM2DeBwcwgfbQWjfW6k9OTQY 1705
kithara/distributed/sharding/_layout.py sha256=GZAqm1OxFQWjLHUdgQob6DbeeUIIH0TSQKK2xw0Z1Mk 1509
kithara/distributed/sharding/_mesh.py sha256=a8EAO5qImV72Gym8WTdJMkkG5vL-hyVv1NAEQ6ltsCY 1653
kithara/distributed/sharding/maxtext.py sha256=jShdT4HoWgCUmHc8VYABiKSQgtlgtXlJWFdui2R7qos 3545
kithara/distributed/sharding/strategy.py sha256=PMWApfDTH-cC0NaUtQTPjHamb9ZebLwbmOmunMCUi04 4231
kithara/distributed/sharding/utils.py sha256=ALS5sX1I1uPei5_dIFidEyT_yuWF6efCWbG4DFe0MO8 6268
kithara/distributed/sharding/models/__init__.py sha256=kdLM5T9psBNFxV1ETcClPDhB4ul1GFhf12LXGHY15ik 67
kithara/distributed/sharding/models/gemma.py sha256=FiZmn0ujcZqBLKEPNZjwmkVRVxuAJkO06jdGn00i7ow 1241
kithara/distributed/sharding/models/llama.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
kithara/model/README.md sha256=tSemZ7vGYLp5ZpEt8H67zGzpKNeWAhGexWF5l0Ctj0Q 2741
kithara/model/__init__.py sha256=o266MyZWUOOujsrhM7g07__mVjyoen9aEW25gsFkIJM 136
kithara/model/model.py sha256=My41Qtl-aGcwPQz4o1hFtruIyEf5lbzG1JlNWO8V_LI 20376
kithara/model/supported_models.py sha256=OuF6SNfSR5W_wdYKiZ1ry91N6qfZL_2BjKG2EWj9Ywk 73
kithara/model/hf_compatibility/__init__.py sha256=1tOu_AyNTuDcl9hiagqdb6ckAqRfR4oQX2iP166ewyY 179
kithara/model/hf_compatibility/model_configs.py sha256=xGpGODT0uVyy30eUcyn5IEjHSBfhD89PtObUla5fEHU 2270
kithara/model/hf_compatibility/shape_mapping.py sha256=UWOi2tHN-nXvaocCGOcVi7F7FmcJIpeFJJxOmzM3F1o 3262
kithara/model/hf_compatibility/to_huggingface.py sha256=dWhOGQvJXlnm01J7YdbnWGk-NfXXXoVsVqQovziYaq8 8289
kithara/model/kerashub/__init__.py sha256=BJ1P9vbNAnkTY-wjaC1OhZynyxwRtb1F36oZtsmv0Kw 52
kithara/model/kerashub/keras_hub_model.py sha256=51b66F00pWT9bJ6ESVMZUQz_msseNOAoRNwuuTvJ3Ys 7571
kithara/model/kerashub/ckpt_compatibility/param_mapping.py sha256=a7Sw0qQgNVXdpY98nc3BKySv4y8cksJfSNKU9mrUrvg 7438
kithara/model/kerashub/ckpt_compatibility/to_huggingface.py sha256=zHgDg2z7kEP4tBiYxHxPFNimLCNHdLnCo4ezdTeE9ls 6023
kithara/model/maxtext/__init__.py sha256=sRe_Jyz9mXJIMBfgP9ozxmvzrMk4zasQBIHOfzM-0aU 49
kithara/model/maxtext/conversion_utils.py sha256=UmWajYyGmXObymsCYjFmHlvXNlerr9LN_O9c5PggS1k 10270
kithara/model/maxtext/maxtext_model.py sha256=HvPgtyzv8iq1n_xwx6e6pcG9OmBUfl1ggCv8H6YqbCQ 8420
kithara/model/maxtext/ckpt_compatibility/__init__.py sha256=85BFL50f_P7vx6IuVi8IGk3Qkz_yHhVnRpFOPecQo7o 271
kithara/model/maxtext/ckpt_compatibility/from_huggingface.py sha256=qvBsiosyW0w0VFH7Lna0aYCckPpEGYbojri0j6Tgvao 5593
kithara/model/maxtext/ckpt_compatibility/param_mapping.py sha256=mmxWpf_dZqEQwFLdbiRpA3OLeh55h6kp5vXg8kwrMiw 19739
kithara/model/maxtext/ckpt_compatibility/to_huggingface.py sha256=xyI5-v7KMkWIFDz6ymYwesj70Ir4PDmA-JLN9gtwIZ8 3402
kithara/model/maxtext/ckpt_compatibility/utils.py sha256=gRw0Km6EyiYj5h7stKwzlaMk5DNAVmyutr2qKQTZCa8 1416
kithara/model/maxtext/maxtext/.dockerignore sha256=_QPxb8Wti-cNbgSR-i4Cm8SjH1sN5tSt-OfdcJ0jhA8 5
kithara/model/maxtext/maxtext/.git sha256=XRqxrocNzLqiKpmHOHqL-BuiJ349MeWdUMzdQwgnA8I 63
kithara/model/maxtext/maxtext/.gitignore sha256=1mIWlE8to0SXwz4qw1vY9lzDTTNKneB9uZVy0EzEV7w 1877
kithara/model/maxtext/maxtext/.pre-commit-config.yaml sha256=Qxq_Df29361SQ_Y11VlhUT69bzh61TMVHLBk8vuSzq4 215
kithara/model/maxtext/maxtext/AUTHORS sha256=kbdMM4KiM3YreLkMyk3BtUmt_wQpBOdL8q5wEw_Fa_w 12
kithara/model/maxtext/maxtext/CONTRIBUTING.md sha256=XVXrj4LuvTRVhQWvtn_TmPADwUZxLm-nT-oS1rXV9l0 1128
kithara/model/maxtext/maxtext/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
kithara/model/maxtext/maxtext/PREFLIGHT.md sha256=9ybBDiZlmOjkTkJrRpFJ1EP-CyXPLS64GZ0c5yD6c60 2529
kithara/model/maxtext/maxtext/README.md sha256=TAXvgRpV2-uikkBRUrrMV-izyRzuQG-0XbQY7pFXlOw 14881
kithara/model/maxtext/maxtext/code_style.sh sha256=T40FOtueQGEE9OX9hudUu5uXqRB8HHuicaeF0RVmYbg 1356
kithara/model/maxtext/maxtext/constraints_gpu.txt sha256=9RYXxQSXgXAY3OA8AkTvfHw8j95C5iP3z0QZJu7tnCQ 4126
kithara/model/maxtext/maxtext/docker_build_dependency_image.sh sha256=ig-dR7X680f_XRSkmJ1Rz72b4mf4ignkgiGgLJzNUMc 5080
kithara/model/maxtext/maxtext/docker_upload_runner.sh sha256=TBPcH6aCRnHMxAhHEhbLGGb1J3IZ8nIEB41BTxVbgdM 2025
kithara/model/maxtext/maxtext/download_dataset.sh sha256=4Klxgj7JWyimfGY7kFwLTyVtf5bB-_-O_VQFzAALGUk 1138
kithara/model/maxtext/maxtext/gpu_multi_process_run.sh sha256=WOe0EPAHGRTmlVgnwPTOvNxTlPzvR-eZbN86qqZl9jk 5166
kithara/model/maxtext/maxtext/multihost_job.py sha256=rkn3VK6elHXen9qWhf7vzLgdy9b9oQqSjZEO0Umrh-k 16352
kithara/model/maxtext/maxtext/multihost_runner.py sha256=JOmRbQlWpniJE0aPgXM1SbjbB-v_q32miNuDQeI8Roc 16835
kithara/model/maxtext/maxtext/preflight.sh sha256=aGApPiMgmo8y7s8YpBwcCrDUvjGcJQqql0VDr3EWJWE 829
kithara/model/maxtext/maxtext/pylintrc sha256=r_U2i437GwSzwfF-m3bOPNVzqyLHzRayxlip-LS8koo 13533
kithara/model/maxtext/maxtext/pyproject.toml sha256=XI4siw5y9DLpQ5buPjNT98kpTGzcbCU49ISIdv23vLY 1172
kithara/model/maxtext/maxtext/requirements.txt sha256=2srJAd9TWM2baNugpG-uRMe82vofwlzYw3tp6FPFROs 695
kithara/model/maxtext/maxtext/requirements_with_jax_stable_stack.txt sha256=K_60z88Hn5HjFOeVEaRo-nijBM_ETQDEjw3jqL94ec0 556
kithara/model/maxtext/maxtext/rto_setup.sh sha256=MFyEVSxSzsySVIcbCY9U5W4otnJrwYK_MYdgpgy1yrc 1524
kithara/model/maxtext/maxtext/setup.sh sha256=obz7umv53PghvYhwF6PdKEfInvLC_pCuojzClQqqBak 7779
kithara/model/maxtext/maxtext/setup_gcsfuse.sh sha256=u4ahykrnooFVlunbsCC-xTFnXQrt_bdo_eq2bQQhPqw 2020
kithara/model/maxtext/maxtext/setup_with_retries.sh sha256=A8ADNkIkmFm7FimVwY91MEsmNzvFXuZF-5ZdSSERmdY 618
kithara/model/maxtext/maxtext/unit_test_and_lint.sh sha256=XleikVG-S7XrljUGyJICU1KzJIII02mJMMhG2u6rQNg 661
kithara/model/maxtext/maxtext/.github/CODEOWNERS sha256=-DMg9AE6n-AOeNvgiyYm_hmlxqgZ-zoq24hU9oZxOI8 162
kithara/model/maxtext/maxtext/.github/PULL_REQUEST_TEMPLATE.md sha256=gBUtwna7aaW-AahDkmfz_d_NNzcRLrRHijz4u5Y8_Cw 1025
kithara/model/maxtext/maxtext/.github/workflows/AddLabel.yml sha256=SV3J3v1EKcllPrIyLK7eaDuviLKLrY3EjW-7LzH-MS0 4551
kithara/model/maxtext/maxtext/.github/workflows/CPUTests.yml sha256=g6SZMDdOMM1_lqkMh-5JDoTMDpidoN1V_2w90gWrspM 1556
kithara/model/maxtext/maxtext/.github/workflows/RunTests.yml sha256=r_lpCppUl-CoLW8pUj9yDrBdbKOjhYeRw7VAJs0tGQQ 4355
kithara/model/maxtext/maxtext/.github/workflows/UploadDockerImages.yml sha256=CKkk70dviHu2p-vg_2lWZKixgQU_a3YtgNzB-6TvgY8 4372
kithara/model/maxtext/maxtext/.github/workflows/build_and_upload_images.sh sha256=o_N6wQsbMSzbJp4kBHBRaWTH_3jnpXlEWfcm2S11OOI 2577
kithara/model/maxtext/maxtext/.github/workflows/build_upload_internal.yml sha256=8qeb1_k7zUF2IpQSdWuelNo6Qblyyimj6ZkZgTmDRi4 1709
kithara/model/maxtext/maxtext/.github/workflows/require-checklist.yml sha256=ZSNFeoGFx4lkrVXV1xYOpwQQuQk-gaIw3W0MlOfhSnE 318
kithara/model/maxtext/maxtext/.github/workflows/run_tests_internal.yml sha256=9rM5SIJylPlZGijdoFVI_i-Hqo0YWQug44pKWWon9hU 1905
kithara/model/maxtext/maxtext/.github/workflows/utils/setup_runner.sh sha256=DhLoYGwRUXLkNVF7VtApZaXKRZwhPDF8-0Oi_hQnhjc 3043
kithara/model/maxtext/maxtext/.vscode/launch.json sha256=zALVmS-HXOoiIRkdIB15QNAbI9FQVfge9RWTXcWwquc 2433
kithara/model/maxtext/maxtext/.vscode/settings.json sha256=JH3QtX5SBiX8lt1tO_qN5jt_EfhrSqtwQxv97fi7Eoo 183
kithara/model/maxtext/maxtext/MaxText/__init__.py sha256=7mu-VQDQMyxM5To0KOhuYe4y2TYGsEkfV7hXZmUyih4 561
kithara/model/maxtext/maxtext/MaxText/accelerator_to_spec_map.py sha256=54V3bY21R5gz7pJTtDzCXgzaYHjgTSNHnI6Cb85aa58 15598
kithara/model/maxtext/maxtext/MaxText/checkpointing.py sha256=1gMOf9GxaQvORDv4iPLUmBVb2TzWlK5C9aNJNNslqgI 12825
kithara/model/maxtext/maxtext/MaxText/common_types.py sha256=lQ-2VCiS53JYt1bdYueMC-wCD75-EXS-CtK1SDtmIgU 1907
kithara/model/maxtext/maxtext/MaxText/convert_gemma2_chkpt.py sha256=-jBzY-3_AFqWb_Au-vu1ssxT6tbpGVu_AeyP0FRwssE 12670
kithara/model/maxtext/maxtext/MaxText/convert_gemma_chkpt.py sha256=TBlN089Qes8cobTGTCRHb-Ald32CEmbm4EzpBS9RJkY 7103
kithara/model/maxtext/maxtext/MaxText/convert_gpt3_ckpt_from_paxml.py sha256=nc5jhEFQQtfmZY3qCCzXAvGhEHzMoMJ8aEVwaQa25pk 12170
kithara/model/maxtext/maxtext/MaxText/decode.py sha256=dIGlSj7DINUFWICNIESOXqyB1_nP8O7WEpB05Wc6JIU 2995
kithara/model/maxtext/maxtext/MaxText/gcp_workload_monitor.py sha256=52wGkCxjt5z3kz35CVMENTkEROMNESFjRYn1ZsdeuBY 8535
kithara/model/maxtext/maxtext/MaxText/generate_param_only_checkpoint.py sha256=UgYYmvEYT-OrumTOitkwsPUOMWopZ3A2fa2f6fK1KJU 6056
kithara/model/maxtext/maxtext/MaxText/inference_microbenchmark.py sha256=smQzKyhk5PEr2RCGwRwfO4SIOzxGrt2aD9i-Owbs3LE 13096
kithara/model/maxtext/maxtext/MaxText/inference_microbenchmark_sweep.py sha256=wQSN-8ToBQX19HSADXS6VNfWPWX3xfmxFIhQkIK1Z6M 6230
kithara/model/maxtext/maxtext/MaxText/inference_utils.py sha256=8oEVENtC8yHKiwZUOuLjnQF-Dst068aFZicrJ9ravDY 3198
kithara/model/maxtext/maxtext/MaxText/llama_ckpt_conversion_inference_only.py sha256=xp0ON7589pHdN_892Ur0RkPhBwcrsXrSU2_dicoJkmo 11322
kithara/model/maxtext/maxtext/MaxText/llama_mistral_mixtral_orbax_to_hf.py sha256=Wtc9Y60HBdEflBPa_lkVPRyMLIef-JxqaLq0E6M_Nsw 10461
kithara/model/maxtext/maxtext/MaxText/llama_or_mistral_ckpt.py sha256=iQ1B8onhM4C8kTTD0E3s8u7r0m9BBOHj36uaRIU4DQw 23221
kithara/model/maxtext/maxtext/MaxText/max_logging.py sha256=hQypDtf-NFe4KrhECUx36A7SGe-TR0BGnnT6Mmn6Z_4 688
kithara/model/maxtext/maxtext/MaxText/max_utils.py sha256=WxOgL99ubJFYwDTE4t7yoBuEIRBRtVjrJDOjPp5IyII 40294
kithara/model/maxtext/maxtext/MaxText/maxengine.py sha256=a2ReEaS7wo5NlyKC2cMS_-h95Pf7w51xT8ntwcJJFmI 32928
kithara/model/maxtext/maxtext/MaxText/maxengine_config.py sha256=if4UVJ0gWx-CY-4DQ9KLoGbwI-IPX7xD33fsQ7oqrG8 2985
kithara/model/maxtext/maxtext/MaxText/maxengine_server.py sha256=XfjDiVMIgyldA2ZJTrqMe4WlS3gAkwwLDFee9Vs9AKE 2359
kithara/model/maxtext/maxtext/MaxText/maxtext_utils.py sha256=sh5KLpGChByk6G5JG_8EPf3OBr-F4Krsxw315lbqOlY 12803
kithara/model/maxtext/maxtext/MaxText/multihost_dataloading.py sha256=ylNpV9ZFHd-Q_-5QKyV7s1dlglVgBqhfL-pcts_koQs 4179
kithara/model/maxtext/maxtext/MaxText/optimizers.py sha256=hNqD7mYsza5Q8-buoKeqP1JjVQG3xC2cE0xjMLxxNG8 5391
kithara/model/maxtext/maxtext/MaxText/profiler.py sha256=uLO12Vx4CcXc341jqNEbLufRT6Q6L2j07WX8ukEBrY4 3808
kithara/model/maxtext/maxtext/MaxText/pyconfig.py sha256=4FYiW9o0eaz1DweVSwmwHBO2O4wmhAbL214uqSb2KcY 35467
kithara/model/maxtext/maxtext/MaxText/pytest.ini sha256=OW_Gg7CRX3waupIinHuw6zOcxwGM9BFirHzlbMP_WR4 643
kithara/model/maxtext/maxtext/MaxText/sequence_packing.py sha256=Jo2dMYb_WHf7djbV-yDfdJ1P28DNHF9vkSXkk5vghck 7674
kithara/model/maxtext/maxtext/MaxText/standalone_checkpointer.py sha256=3hEoOs8wcHzJnaitxqYwMQgMUe3KHp9rxeVqW40mN-M 4507
kithara/model/maxtext/maxtext/MaxText/standalone_dataloader.py sha256=pInTqdvMrjk8Lfkyzli2PkleCliuHiQZPlQN0o0bSt0 2741
kithara/model/maxtext/maxtext/MaxText/tokenizer.py sha256=5lv5YYsgFmEndHeyLhvHcxhqHJVUZUvElJrqNq4UXR4 7737
kithara/model/maxtext/maxtext/MaxText/train.py sha256=A3p3VxAKPj9Ij3HXAefgUB1OSZeiQmq4dhbIkuFk7IQ 42221
kithara/model/maxtext/maxtext/MaxText/train_compile.py sha256=EEyg95ooZJUQZIpJXdc7imIZCIkkBJ-yZkXc29XnEOc 7280
kithara/model/maxtext/maxtext/MaxText/train_tokenizer.py sha256=5H8RmGJYA3aQGQWMx1CCn9nxV9WoxA5j-DG9bCn7MHI 5491
kithara/model/maxtext/maxtext/MaxText/vertex_tensorboard.py sha256=96lTvvYKSOecYn8uPL-REipVGX0NSWgsjPgCoQFwS48 4965
kithara/model/maxtext/maxtext/MaxText/weight_inspector.py sha256=iB0oZQZOVkzQmZ2h9KffFsPImZJXyrKUUDpd0Je-shQ 2408
kithara/model/maxtext/maxtext/MaxText/MaxText.egg-info/PKG-INFO sha256=U5JiW2ESJJQf0Fb_kosCspBuIJZSTdIV8tfnDLCPioc 29466
kithara/model/maxtext/maxtext/MaxText/MaxText.egg-info/SOURCES.txt sha256=tOUqDU39PSLq1rZjztXwIs3wGmiJcwk_NZuu56QcR5c 251
kithara/model/maxtext/maxtext/MaxText/MaxText.egg-info/dependency_links.txt sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs 1
kithara/model/maxtext/maxtext/MaxText/MaxText.egg-info/requires.txt sha256=vny_EEDSAZNhjQdnOcaipQIDU71bgnSUKPMnF5chRCg 697
kithara/model/maxtext/maxtext/MaxText/MaxText.egg-info/top_level.txt sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs 1
kithara/model/maxtext/maxtext/MaxText/configs/README.md sha256=xo5Z8Tp_0kcZC2adiFr2Bh3zaiatOnCDaVnVxewgih0 19330
kithara/model/maxtext/maxtext/MaxText/configs/base.yml sha256=sFdnoQKA-wj_fK75YGIfu3xlSSb0TIJmyaYftMm3EGo 28607
kithara/model/maxtext/maxtext/MaxText/configs/dpo.yml sha256=OjQdBKnY9lLpH-OinZEfa8sOt3qJI9CmB0cs2Gd0hs0 757
kithara/model/maxtext/maxtext/MaxText/configs/gpu_smoke_test.yml sha256=nhf-UgpyGBoo4OfFqMkbn3d6OqZCnQQ51EAcVihFf7o 285
kithara/model/maxtext/maxtext/MaxText/configs/inference_jetstream.yml sha256=xVnFEvb4U30YXxe8_qYLzPKzLQy8ZvrEmBrO5W6ll1E 103
kithara/model/maxtext/maxtext/MaxText/configs/tpu_smoke_test.yml sha256=q20Ea-ytIf0blhjY0AbHSCE7dL3e9RukHh6odt61xZY 273
kithara/model/maxtext/maxtext/MaxText/configs/a3/llama_2_7b/16vm.sh sha256=MJHsJJLwzoE7W3q8tpACCwkyPKCzfEYeNAj8s6Ke6_s 1610
kithara/model/maxtext/maxtext/MaxText/configs/a3/llama_2_7b/1vm.sh sha256=BJz1mZsccIwDZNqCyIbZLUVGInCG29Oh2akKac71Stw 1585
kithara/model/maxtext/maxtext/MaxText/configs/a3/llama_2_7b/2vm.sh sha256=vztKFDUAmSMQPn9as9-UwewRMcz4awJdZNaqJgYFwgQ 1554
kithara/model/maxtext/maxtext/MaxText/configs/a3/llama_2_7b/4vm.sh sha256=aXvDLbUYL2Ie4on1Bp2Jj1eo3bSCelRP1pTHKQj3hSQ 1554
kithara/model/maxtext/maxtext/MaxText/configs/a3/llama_2_7b/8vm.sh sha256=pywJHoji4TGy1HLIFkYdPvif-PTYy-KRkRpdMgQv1u0 1555
kithara/model/maxtext/maxtext/MaxText/configs/a3/llama_2_7b/README.md sha256=TMOvc09F1RF80H35GiZL61dk43Y_1Xzh8tdTkFA3O_M 1023
kithara/model/maxtext/maxtext/MaxText/configs/a3/llama_3.1_405b/128vm.sh sha256=rSoccuQ1eN_1-AsW2wm98a196Jyy2QVtNn8bIxiUc0Q 2243
kithara/model/maxtext/maxtext/MaxText/configs/experimental/1024b.sh sha256=5_WI0UAnUappieg5PPJnf4SPZ1rv_WgL3K3jerjaPAw 1399
kithara/model/maxtext/maxtext/MaxText/configs/experimental/128b.sh sha256=ADRiPQa6WKDT_rzt6jKDJMH4X3Y9X49-5mZTVj6XuHg 1399
kithara/model/maxtext/maxtext/MaxText/configs/experimental/256b.sh sha256=_Esyqi4JIW69IAhYo0l74IR1T3tZxrAYOeXR882b8OA 1398
kithara/model/maxtext/maxtext/MaxText/configs/experimental/32b.sh sha256=PLo6ziPYASmsJgRrRBrdVrg7xHS8pdXsDho6W8YxopI 1396
kithara/model/maxtext/maxtext/MaxText/configs/experimental/512b.sh sha256=RxvzbF64fsivf_Rkme868PE_hoZaSaeIh0FCIe7_y94 1396
kithara/model/maxtext/maxtext/MaxText/configs/experimental/64b.sh sha256=Y-T5nz8SwkYOon_gii4FUvPA5vszJ9hvCQozgPQIIiU 1396
kithara/model/maxtext/maxtext/MaxText/configs/models/gemma-2b.yml sha256=FJSHI3LVymWh3eCsaAZLDCcrq-RXZR2Oa8oc7FiMNPc 871
kithara/model/maxtext/maxtext/MaxText/configs/models/gemma-7b.yml sha256=weUeP3f3shIef_isXpY25i6gTESXlLjQeuF0wZiGo5A 873
kithara/model/maxtext/maxtext/MaxText/configs/models/gemma2-27b.yml sha256=o-CokfLa2MVKGT9OLm7F7xQABLG5m8FhBUjoaij7E-U 1112
kithara/model/maxtext/maxtext/MaxText/configs/models/gemma2-2b.yml sha256=C7seC3QMZZ6wYf_AfIFLR1WpSc93oAZx4eGXyb7ilFg 1107
kithara/model/maxtext/maxtext/MaxText/configs/models/gemma2-9b.yml sha256=pt372xq7qI2zQAZTUoGA220-8ogADZPiFlboKNEiVoI 1110
kithara/model/maxtext/maxtext/MaxText/configs/models/gpt3-175b.yml sha256=HXYixk0rdUkdRY82JPljJdiUQzPkT_gb6UvkRtMf3Vo 1312
kithara/model/maxtext/maxtext/MaxText/configs/models/gpt3-22b.yml sha256=nhLwEG3M0rnMS0E_IMoR1cZU5gqTo1RnCAgi-BCLAXU 1152
kithara/model/maxtext/maxtext/MaxText/configs/models/gpt3-52k.yml sha256=KPVzlKkVPugXAvkPxxyhUUPzPm2yWLUlAAGBE7cb32Q 1249
kithara/model/maxtext/maxtext/MaxText/configs/models/gpt3-6b.yml sha256=MlClTAntM-_M6tNrDxKReSmFoH4usah_IkX5vekR7Ig 1151
kithara/model/maxtext/maxtext/MaxText/configs/models/llama2-13b.yml sha256=B-P3LoBDaWqwDzdl_IgpJTKrrbNZ8andiulL3ljG7B4 938
kithara/model/maxtext/maxtext/MaxText/configs/models/llama2-70b.yml sha256=E6U0ZD1_JOPiuuOcBw8ezPatjJioq6_eu7rVUzyEhj0 914
kithara/model/maxtext/maxtext/MaxText/configs/models/llama2-7b.yml sha256=9YMbAanSPknuzi3oP1QBbwNw3HDWMIBUSrOPaqd0uu4 937
kithara/model/maxtext/maxtext/MaxText/configs/models/llama3-405b.yml sha256=8w8jPqwU1iHgMl_mcARG6rkMgSNyVGLucYHIebESbPw 972
kithara/model/maxtext/maxtext/MaxText/configs/models/llama3-70b.yml sha256=zX59jqWjGwhT2Fz5jHEi26VH8D97v0nyMuZvZ1-HAbw 967
kithara/model/maxtext/maxtext/MaxText/configs/models/llama3-8b.yml sha256=uzsSJAEvSuiIA1DvLNYoOpRDNtXs8w2ENZXkGtVDwcA 966
kithara/model/maxtext/maxtext/MaxText/configs/models/llama3.1-405b.yml sha256=LejtmU0XaF2zUQSJv-SJZa6WfsUCOScxN1bpxC6sXdA 973
kithara/model/maxtext/maxtext/MaxText/configs/models/llama3.1-70b.yml sha256=lgP-bISwvYrowaO6H7pL_C6vDIUPfE5w6MddG_58M-I 969
kithara/model/maxtext/maxtext/MaxText/configs/models/llama3.1-8b.yml sha256=BPXDX_oPOA1pvv5p2tIH3v4KfJnyxguUYQi10mcBvow 968
kithara/model/maxtext/maxtext/MaxText/configs/models/mistral-7b.yml sha256=t-xWmtMU-JWqcmIipP5qcnKnFH6PZB27-lYr0ix0G4w 929
kithara/model/maxtext/maxtext/MaxText/configs/models/mixtral-8x22b.yml sha256=hGCm-FFuoFDydYXCwyPlIfHUxSCZHxCMSu6xddWDmxY 970
kithara/model/maxtext/maxtext/MaxText/configs/models/mixtral-8x7b.yml sha256=OnbOqGv-rCLJnocMTBFfgx-l7vH-uZl9c2n9a_gt2ec 969
kithara/model/maxtext/maxtext/MaxText/configs/models/gpu/llama2_70b.yml sha256=dRvb3EHWl9DhMmX1-xQitXzT3q7rRAAQzTSRLcUkOT0 417
kithara/model/maxtext/maxtext/MaxText/configs/models/gpu/llama2_7b.yml sha256=_VfJkn7FtqakOHY9O54FvpszZZziarX4gSWmFFSryvk 329
kithara/model/maxtext/maxtext/MaxText/configs/models/gpu/llama3.1_405b.yml sha256=EJqBBSRu9dvIw6PmgBSUOW_zBQcg6CbHwnyNF65On1U 421
kithara/model/maxtext/maxtext/MaxText/configs/models/gpu/llama3_70b.yml sha256=lC3kjuXEcFYzY0U94PKOp0WxwdIIbZDkDs0psYb_Yow 1021
kithara/model/maxtext/maxtext/MaxText/configs/models/gpu/llama3_8b.yml sha256=oz2dOrOzK838Q0UYYpVZdwV5-8g_kBtfHEeXYKPbOyU 1029
kithara/model/maxtext/maxtext/MaxText/configs/models/gpu/mixtral_8x1b.yml sha256=rXPChXG4nLX4Kwzc8xOnw3sq9IRZj4XN4FAPoJck5v8 1113
kithara/model/maxtext/maxtext/MaxText/configs/models/gpu/mixtral_8x2b.yml sha256=pVO0lGsp3G_AcdGBnVofCCpSvW3DLqROiLm0MldP7PQ 1114
kithara/model/maxtext/maxtext/MaxText/configs/quantization/README.md sha256=VtdOXWMn1iPv-Wbpa4JEEKFHNS3ka0xAxO3hL2J2KVM 1471
kithara/model/maxtext/maxtext/MaxText/configs/quantization/dense_llm_subchannel.json sha256=0xla-5pX3LLFnZxZ6kpYZ8DKnSAGxL6gToWm3qf2V98 250
kithara/model/maxtext/maxtext/MaxText/configs/quantization/dense_llm_weight_only_scale.json sha256=ivPXJ5ZqRA0umVpYDBYepD7OgeIsv0dBhBlaEd0yFuQ 232
kithara/model/maxtext/maxtext/MaxText/configs/quantization/int4_weight_only.json sha256=oKNQduLQF38glRTzYVfH2tpFKOCE6StGJIgP_U3CLJA 34
kithara/model/maxtext/maxtext/MaxText/configs/quantization/int8_weight_only.json sha256=75u1lxq4LHGumkoOSZwckwcOgIUE4H431jAmfVWOFuk 34
kithara/model/maxtext/maxtext/MaxText/configs/trillium/gemma2_27b.sh sha256=0Vkp1JV_bbRsTJ7qXeZLddeVYZ0rsrDH4SumM4hKZuc 2005
kithara/model/maxtext/maxtext/MaxText/configs/trillium/gemma2_9b.sh sha256=30-v28XdjqlRXmZKoUJf3rLd2gJAMO8subSCjNNj_1A 2001
kithara/model/maxtext/maxtext/MaxText/configs/trillium/gpt3_175b.sh sha256=Yr4byBTVHGw0DjWwjb8vHeaJa-mJF-A1TfPc2EnPu8w 2018
kithara/model/maxtext/maxtext/MaxText/configs/trillium/llama2_7b_4096.sh sha256=1pyydeFk7UQxyGDwGtXeqokFORS537soAkeGl7D2eA8 1855
kithara/model/maxtext/maxtext/MaxText/configs/trillium/mixtral_8x7b.sh sha256=tOybHrr2gZt88cJKEC71LSo3gpFE9xgnYMvUsFSfyCI 1859
kithara/model/maxtext/maxtext/MaxText/configs/v4/22b.sh sha256=an2sWr5ZAdUdfcH6V2x6wGGs-tLflA6PArS8a5pFcww 2315
kithara/model/maxtext/maxtext/MaxText/configs/v4/52b.sh sha256=zVHw7BNxHbW3nmZsLmTvPgqOkUkOhuKwhWrGsX9ssX4 2378
kithara/model/maxtext/maxtext/MaxText/configs/v4/README.md sha256=WtHryFIcTT8JAGjdvNHPAmTPe6p0uR-y88kXUy0Y6cs 1273
kithara/model/maxtext/maxtext/MaxText/configs/v5e/128b.sh sha256=y2E6KyPDB74OKDllLDs26zgcCa2seF6WuaV5Sonz1sE 2143
kithara/model/maxtext/maxtext/MaxText/configs/v5e/16b.sh sha256=UbaMSYy2GMSF2PinzrEpSz5YMalD0bKzOz8ZSLQsh5Y 2055
kithara/model/maxtext/maxtext/MaxText/configs/v5e/32b.sh sha256=Wc-nyAkX9zsc-phEhpeZBvR5leDktn0Um3LJSG71C_Y 2054
kithara/model/maxtext/maxtext/MaxText/configs/v5e/64b.sh sha256=p2MkPMTpnK7NfNlV5nnx5ZG0NB7W-j9u2x0WNmvJ6BQ 2054
kithara/model/maxtext/maxtext/MaxText/configs/v5e/README.md sha256=fhzuWrMyjFbTTcb8R53_CXC0BVtZksYE_SARyzko448 1859
kithara/model/maxtext/maxtext/MaxText/configs/v5e/gpt3_175b.sh sha256=yR9rBFyUBwRrtoH2Xu7rnbkNw_AbWUeo0ZrhFRAIllU 2077
kithara/model/maxtext/maxtext/MaxText/configs/v5e/llama2_13b.sh sha256=uPH40TgbPVHYfpo9eK6jmCQ7JV5nBhcN442ArRzbgHg 1975
kithara/model/maxtext/maxtext/MaxText/configs/v5e/llama2_70b.sh sha256=1B2hKVf3-8hMiOCnNTcLgeCUALEKS0ZcYnNcdyYeAr4 1975
kithara/model/maxtext/maxtext/MaxText/configs/v5e/llama2_7b.sh sha256=xTyfQ6JZevAh5i4-INydO0LW-2EVfi_3-oJsonjDk9U 1965
kithara/model/maxtext/maxtext/MaxText/configs/v5e/inference/llama2_70b_v5e-16.yml sha256=y8H2JN8lkSJppFgf7ObjF0NeEL7chZeGkDU7WDBeD1Q 1922
kithara/model/maxtext/maxtext/MaxText/configs/v5e/inference/llama3_405b_v5e-64.yml sha256=H24AF4lrrp5313bBhYteKk1gbTHcySMBAaxGOWgy6_Q 1827
kithara/model/maxtext/maxtext/MaxText/configs/v5e/inference/llama3_70b_v5e-16.yml sha256=W_RtU8W5MEZGT7K2MXQNnDOJd0V5ljy2f7iYEOUDJqY 1973
kithara/model/maxtext/maxtext/MaxText/configs/v5p/1024b.sh sha256=XZ_vTzb6CvRr70lPNqEX2fODT5QasYddeMhHqPCMgx0 2279
kithara/model/maxtext/maxtext/MaxText/configs/v5p/128b.sh sha256=pTloDBUZz03DUP9Q0HzK32-LU64yMyc9TXN30-Fpiks 2274
kithara/model/maxtext/maxtext/MaxText/configs/v5p/256b.sh sha256=fSDUHxaalt7qQPAQLKiW3LTd6Lkm83BRiW6_qGcsris 2327
kithara/model/maxtext/maxtext/MaxText/configs/v5p/32b.sh sha256=5_IoUp6XxrpJoYukGN5aoF9i7JArrwf07FxfS0ehY4g 2470
kithara/model/maxtext/maxtext/MaxText/configs/v5p/512b.sh sha256=lh5evetuLYVZw972EYXFJ3jJ3y4HxZnFo7_40PCFPEk 2324
kithara/model/maxtext/maxtext/MaxText/configs/v5p/64b.sh sha256=CuMnvc_jYvfxB1b700m8JGTFjOKo2Lxbq5S1uefdsHk 2258
kithara/model/maxtext/maxtext/MaxText/configs/v5p/README.md sha256=Wiva6BIiaZ_O0gevHeZVXTkGiIFhfeeyy_ZXfdVcDpE 2148
kithara/model/maxtext/maxtext/MaxText/configs/v5p/llama2_70b.sh sha256=yGvDakISZvokeOWAoZx5n5bvCztLfNu2dJHNom7pv6Q 2452
kithara/model/maxtext/maxtext/MaxText/configs/v5p/llama2_7b.sh sha256=dib3O_xMbYX5gefRAp0eo5dUwXRzjt4TuVx5kjvs9vM 2522
kithara/model/maxtext/maxtext/MaxText/configs/v5p/gpt3_175b/gpt3_175b_base.sh sha256=0KnKxOzRGvuygdjsUe5gtLHLVufnS5AeL76ZiSDBqdQ 3743
kithara/model/maxtext/maxtext/MaxText/configs/v5p/gpt3_175b/v5p_1024.sh sha256=kTNoXZX-ho5XdxNbnR70p-EHs_X2WgOy9eHCr3LyFZQ 532
kithara/model/maxtext/maxtext/MaxText/configs/v5p/gpt3_175b/v5p_12288.sh sha256=yZ_VUmGKc9b3TdCMa7k_fMKG0aAYFWEk2kOwSw0tIBU 573
kithara/model/maxtext/maxtext/MaxText/configs/v5p/gpt3_175b/v5p_2048.sh sha256=Kpa7B5kFInMPUH38uFJMz_VnbqcV9g3ho1tWOL8W4sI 550
kithara/model/maxtext/maxtext/MaxText/configs/v5p/gpt3_175b/v5p_3072.sh sha256=zgvKJsVHNGgN2gzeqW99W8AcapS3EixCfB39XL0tlCE 551
kithara/model/maxtext/maxtext/MaxText/configs/v5p/gpt3_175b/v5p_4096.sh sha256=JsxmXNj7PC7TbBiMsunga7JVz_fiCzoxLowkc-93a00 569
kithara/model/maxtext/maxtext/MaxText/configs/v5p/gpt3_175b/v5p_8192.sh sha256=09LEmBjyEKQxPo7LO0DnbD6B3joNqx_u7SVg9L5wGe4 571
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/README.md sha256=LkkyI1W0MlsfMc-aC4SE7cPZQ69cllmVBiFNAP5iFXc 6773
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/evaluate-accuracy-fast.py sha256=Xebdl0kM5-irsTB2EQu7_kk-CUBk5DVO4C1EIQ-qqNc 6901
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/evaluate-accuracy.py sha256=_J26W8IwwJpZdEwT6m5BRMTXn8SvN_VdzfV7JcnUxNQ 3886
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/llama_offline_run.sh sha256=phyju93ZiXVJOuHBUgqtlzOEftUqCJmTvU7NN676X_k 5448
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/mixtral_offline_run.sh sha256=TkfGBxNUmjsZfTxPwEFgmo-GqR-bCq3wLBqU0aKNVJc 5759
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/offline_inference.py sha256=elexFv4RAyJxeGLHDPHJDXm2fcjEGxrbaYwwfFhiyJs 17007
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/offline_mode.py sha256=mB2bR9qVf1YLdJjYRqhcfs619b821EeV9WLotdSc9Qs 16947
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/requirements.txt sha256=uaZ6yQxF6Nb68QH9PbqIl6mqE0JK1cfWpIyVJkcIARE 149
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/user.conf sha256=tzhHLB7VQU4XrbkiRq_ZqzWLZIyaps8WdHcPEpdhFm8 864
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/user100.conf sha256=lJn2hnEBqpIIz5cf603ltLHq7GoTXfNnidZb0MxV-v4 986
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/user5000.conf sha256=9nnodJ35UH2hB9YDhxsbURv34zWpQfBS1A145ohwRCo 991
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/matmul/matmul_dtypes.py sha256=s7c4zgzzU1FZf7uWdVU2whV50EKxKqgA8d8btBLkcNQ 1377
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/matmul/matmul_sharding.py sha256=M0pznwcyxjqLUxq_xufAU_Ah6IRRcZtqa8C9nNroHxA 2877
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/matmul/timing_util.py sha256=GwT4RA6-4vGfTcfm5yvx7xmNqKXtvmHaj5E5VgUSVRg 2366
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/trillium/benchmarks_llama2-70b-trillium_2x4.sh sha256=T98XBT_h8E7PINZiswdu6_FbvVbib7zBqYK95Szb5FE 3550
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/trillium/microbenchmarks_llama2-70b-trillium_2x4.sh sha256=HwEaUv8e6M8D4wpdqYf29qP0dQDeaLDtD-xMZ-a2tQg 2762
kithara/model/maxtext/maxtext/MaxText/inference_mlperf/trillium/select_xla_flags.py sha256=tdKsWNdXOFdZPna2ln_wFXO3a6lylmLn4Z_7gYm5XOA 3533
kithara/model/maxtext/maxtext/MaxText/input_pipeline/_grain_data_processing.py sha256=3e_AmeeHguujLJon-bcpc1m3KAkP_E4gnqqEXPSvdvw 5386
kithara/model/maxtext/maxtext/MaxText/input_pipeline/_grain_tokenizer.py sha256=1P8f-LAIx8G0tv5z4YyGy4_CmBhwJihjKFiQmEVYD8g 2620
kithara/model/maxtext/maxtext/MaxText/input_pipeline/_hf_data_processing.py sha256=XXL30Ad41k6KIMKRE985H3v8vj76lgSlXrjruQevlNs 6708
kithara/model/maxtext/maxtext/MaxText/input_pipeline/_input_pipeline_utils.py sha256=kDNM3TlrDLXpIJtABtRF7JuQ6qVqqloPsk3_yVfR_no 9680
kithara/model/maxtext/maxtext/MaxText/input_pipeline/_tfds_data_processing.py sha256=lveLNp6P_tQ1WYAMGSprdfO2QtFSyvlodJ1b42CQWlo 7535
kithara/model/maxtext/maxtext/MaxText/input_pipeline/_tfds_data_processing_c4_mlperf.py sha256=KSYj9oXTj9tPm65ubij6DVoZMYzxni9IYb96XeHrcWM 12676
kithara/model/maxtext/maxtext/MaxText/input_pipeline/input_pipeline_interface.py sha256=wlEDw7H_fPKiq6ntjr3_zDE2bX4MACgFviV5ycN1Bu4 8921
kithara/model/maxtext/maxtext/MaxText/kernels/ragged_attention.py sha256=vxG8JxySCcxhSO6ju3jToSx6wQVRn5E3cDW57KD4bwg 13430
kithara/model/maxtext/maxtext/MaxText/kernels/megablox/__init__.py sha256=lfoZdDIyvCeyHz1qP8K9NhBGAGaMiZ1RiYRr0sxTL6M 635
kithara/model/maxtext/maxtext/MaxText/kernels/megablox/common.py sha256=l0psxxQlIgQ27_9CglqiyIohAiSi3YeaeqgmvQeouQs 1933
kithara/model/maxtext/maxtext/MaxText/kernels/megablox/gmm.py sha256=BB3UUusHzoLkG019r-QTKLx4M7pe4tj4TeF8GI7AF6I 30653
kithara/model/maxtext/maxtext/MaxText/kernels/megablox/ops.py sha256=DTceufaHm78_3CbzVKNjxN0nQusLu0z3kF-S5JtiV-A 3449
kithara/model/maxtext/maxtext/MaxText/layers/attentions.py sha256=zxdauNzap9mh6x91XG4fMWW_IQcRw2DMUtUfuTAC7Bk 54491
kithara/model/maxtext/maxtext/MaxText/layers/embeddings.py sha256=n4MC3EbSQ7yetJO-xKXOp3ke97_-P4mL8DkK0ECyvlo 11339
kithara/model/maxtext/maxtext/MaxText/layers/gemma.py sha256=lVC8rtImtAi4cKSY5NA4bwGoyEvVwbi5-m4_PZYedpo 4963
kithara/model/maxtext/maxtext/MaxText/layers/gemma2.py sha256=Ah_uRp3iMmk_NDThOC1UV1yJD4D-4d31dOBwZiiAbII 8515
kithara/model/maxtext/maxtext/MaxText/layers/gpt3.py sha256=y6ipJIxAAYB2y1UUPMK_Y301Ic60xxfi9E2A2L86k7Q 12019
kithara/model/maxtext/maxtext/MaxText/layers/initializers.py sha256=3xFaP_YTo-MvWXua6gRciHHLe5gKC4frpTJXRcOYjmc 1467
kithara/model/maxtext/maxtext/MaxText/layers/linears.py sha256=V7yWpNAaNUjAAVGyU6gEtwyDCX-44NlnqpfoDOGU614 32289
kithara/model/maxtext/maxtext/MaxText/layers/llama2.py sha256=Gpj-zvgHS84K2YWEJtFYVZz1qVD5sfhhA_3AlPzoWv0 5709
kithara/model/maxtext/maxtext/MaxText/layers/mistral.py sha256=DXnE1UJ6km5S-YUpUxcAjAruDlq84OhOSEGeVE7wJMM 5809
kithara/model/maxtext/maxtext/MaxText/layers/models.py sha256=QRhrHDV6nYBgb5N10xZXoWtsK_9uWUq5sLNkslBSUA0 17824
kithara/model/maxtext/maxtext/MaxText/layers/normalizations.py sha256=-yJEiiervbR-tqOnCLUUKwscFWomPcjpgrRCoam3Icc 1583
kithara/model/maxtext/maxtext/MaxText/layers/pipeline.py sha256=hl27v2tAM8Y4Ne9n6-PBtytBkNchV7iP13fPhqD4AFU 36369
kithara/model/maxtext/maxtext/MaxText/layers/quantizations.py sha256=S8OOyVMNvXxCGP5tx3Pc5V9dET9Tcr2zKbB-pabceVo 15438
kithara/model/maxtext/maxtext/MaxText/layers/simple_layer.py sha256=JiLPZySKiNRrB5imsoGqhzPMqy8oxeTeHAQUr2j2pio 2451
kithara/model/maxtext/maxtext/MaxText/scratch_code/analyze_sharegpt.py sha256=p3RpM-ss2R9o6bPna2eNJMFOrzjkSmFSZe-dxbjcK8Q 4662
kithara/model/maxtext/maxtext/MaxText/scratch_code/gemma_7b.sh sha256=qI6z6yDraJuWuydHltwqrTyFRitXdDjODUDDaONFI_U 762
kithara/model/maxtext/maxtext/MaxText/scratch_code/golden_gemma-2b_export.ipynb sha256=Hssex46EhhctzKJ3CucAectadm2huO6veCM9HA5sAaM 6757
kithara/model/maxtext/maxtext/MaxText/scratch_code/golden_gemma2-27b_export-flax.ipynb sha256=waevJOe33Xy2kKYrB_bHaN2NGX7sa5aS3CMgt2MoKwc 15714
kithara/model/maxtext/maxtext/MaxText/scratch_code/golden_gemma2-2b_export-flax.ipynb sha256=ROAieDZ-XxuEbMkk8qFmCPXwqUoalbW_OsaEv-uaeX8 14031
kithara/model/maxtext/maxtext/MaxText/scratch_code/golden_gemma2-2b_export.ipynb sha256=dnK8FnJ2xq1E08cxbQkm3JlwEF-nXcFC4ZQtRNEbb-k 14120
kithara/model/maxtext/maxtext/MaxText/scratch_code/golden_gemma2-9b_export-flax.ipynb sha256=D-oPxGZlpS53CHjyoSdfBfViNXZnS5JDY16_cxctfBs 15323
kithara/model/maxtext/maxtext/MaxText/scratch_code/golden_gemma2-9b_export.ipynb sha256=dWKQUfvSw6FDcBxRB_Gx5AQ0_N2uWkuh_9vPAOcmNnw 13923
kithara/model/maxtext/maxtext/MaxText/scratch_code/golden_llama2-70b_export.py sha256=SaJT4QtEG3zCT0tPvCGJ1TY-AJ83aPXa__SCMR6sN2U 2222
kithara/model/maxtext/maxtext/MaxText/scratch_code/golden_llama2-7b_export.ipynb sha256=VSEbNJZ0_bA7iGzsowFmhapmY4Y9ecK_gcBgX1wWSIo 2967
kithara/model/maxtext/maxtext/MaxText/scratch_code/golden_llama3-8b_export.ipynb sha256=5Nx6TF6QahItaZRp-r9iAx41d8JVURMm6Yhc0IS63XQ 2820
kithara/model/maxtext/maxtext/MaxText/scratch_code/golden_llama3_1_export.py sha256=OomkX8dxQda7JWGYyc7l9vBKw4TQG0ZXa6ooVwJKkR4 2810
kithara/model/maxtext/maxtext/MaxText/scratch_code/golden_mistral-7b_export.ipynb sha256=XlmtC_R0jtBwYSKFKy1pEAL55oLp8OUvvLFDKF7i5vo 7531
kithara/model/maxtext/maxtext/MaxText/scratch_code/golden_mixtral-8x22b_export.ipynb sha256=aZ_psSjSy0Yif9VRB5zBtac2yhFbUyL1mDThpWWuPDk 4214
kithara/model/maxtext/maxtext/MaxText/scratch_code/golden_mixtral-8x7b_export.ipynb sha256=4yGZ2FD946SwBNQcG03_EvAbGB8-bs2GUF6900T1waA 3029
kithara/model/maxtext/maxtext/MaxText/scratch_code/mixtral-numerical-verification.ipynb sha256=m5Y5koOEI6x4lItqHEUVblNtwCkp2QggUuYbSiQLG2o 10022
kithara/model/maxtext/maxtext/MaxText/scratch_code/run_inference_microbenchmark.sh sha256=Cxhut2Q_It07OMPx4PPfy4axyjATQqi9AMh9aiQTzmo 483
kithara/model/maxtext/maxtext/MaxText/scratch_code/setup_transformer.sh sha256=xXomzL1zrkjZPEcZ-BO8IA5bN8NgaNhIgfjJ8egIL7U 159
kithara/model/maxtext/maxtext/MaxText/tests/aot_hlo_identical_script.sh sha256=Ycsw_o2HOmByahfR-oeMF5dtm22TGc8-Nie8VLwwDOY 1346
kithara/model/maxtext/maxtext/MaxText/tests/aot_hlo_identical_test.py sha256=EjN9WvGKivk2R7dNInIX1A3QgCctaM0ST-HdNVqiKws 4481
kithara/model/maxtext/maxtext/MaxText/tests/attention_test.py sha256=YPJh1p5DpYKIR_7YsNWZcNjz5PJBylJpGwWQgUlC0U4 26583
kithara/model/maxtext/maxtext/MaxText/tests/decode_tests.py sha256=4RB47zkywwJmVYfdhIqy22FhEtwWCkvSf-3d414J9TE 3046
kithara/model/maxtext/maxtext/MaxText/tests/forward_pass_logit_checker.py sha256=HVJLE9qak7upw6LYKmSEfK_PRafyAMsU6dzBdYngXoY 8125
kithara/model/maxtext/maxtext/MaxText/tests/gpt3_test.py sha256=zzZug1pUpoDFLEjRHBJVtT9nTduv9arZP9rBf7whpKM 4235
kithara/model/maxtext/maxtext/MaxText/tests/gradient_accumulation_test.py sha256=bb7LDj8QgsCqUv3VxXgnRKeQaCZ3SD0LG-reF2r3BYY 4887
kithara/model/maxtext/maxtext/MaxText/tests/grain_data_processing_test.py sha256=ZzcBmUfT6ThFkMv30SY-xcuRQ13xkC36c2t9sgidpYM 4269
kithara/model/maxtext/maxtext/MaxText/tests/hf_data_processing_test.py sha256=kPTiMRYGG9_h4GptLe_3mcDj_5aHRpf2QnDCVrR3IFs 4278
kithara/model/maxtext/maxtext/MaxText/tests/inference_microbenchmark_smoke_test.py sha256=8DseYxeAy8EjIMwUPOs8PBIDytg43dBvOmpRLHVcs0w 1329
kithara/model/maxtext/maxtext/MaxText/tests/kernels_test.py sha256=tl2lRRdhdQYE9iPpF3T6SBR7bmZMzbSrHhMWH0kzSpk 4397
kithara/model/maxtext/maxtext/MaxText/tests/llama_test.py sha256=S5JuMfBqhMG3z3rPzRGP0zrufSOFUVW2_oxtB-fOcr0 7645
kithara/model/maxtext/maxtext/MaxText/tests/max_utils_test.py sha256=nDxptfqCqsdYBrIzYZe1gAZJ0-ly-NQj1w8gtvXn6Cc 8479
kithara/model/maxtext/maxtext/MaxText/tests/maxengine_test.py sha256=aDI0UtDFmy7myFyTx-6bNCfY7pjeEGRCnPA7s9NtdoA 4115
kithara/model/maxtext/maxtext/MaxText/tests/maxtext_utils_test.py sha256=d5ZBXvW_Bi5FXdYxDTvOh4vcuqy21x3mLnNdrrs30ec 3217
kithara/model/maxtext/maxtext/MaxText/tests/model_test.py sha256=xcAOFGaZactLGKEhJfL5O1BtyWYKsNKUMP60vNrKpKM 5594
kithara/model/maxtext/maxtext/MaxText/tests/moe_test.py sha256=smQg0vPk0MpQl5C-NiPpVpcm3qF9KVT5GCBuh_H6G8c 12383
kithara/model/maxtext/maxtext/MaxText/tests/multihost_dataloading_test.py sha256=0dSrihxfObhUQA0VRLxQ9aukdXfQX5Ymo7IEleHEV0k 2342
kithara/model/maxtext/maxtext/MaxText/tests/pipeline_parallelism_test.py sha256=zDvqSrvzwNNEyCLc9KfNhjGuNfxekFYf2F4Yiy4iWCc 13236
kithara/model/maxtext/maxtext/MaxText/tests/profiler_test.py sha256=3tc_WLuJQuchM3e4FUd4ZXAmLuf99abJO33RPdPGv7w 3361
kithara/model/maxtext/maxtext/MaxText/tests/pyconfig_test.py sha256=bBI4ZabUXlMknuRbFvPwbke8skX0a-x9t7v9RMeat-0 2148
kithara/model/maxtext/maxtext/MaxText/tests/quantizations_test.py sha256=nosZZbDZLcS3XZP_pJWoOrZc89gc0I-DmVKx3crHtV0 8575
kithara/model/maxtext/maxtext/MaxText/tests/simple_decoder_layer_test.py sha256=6jK1ZbNY4FFgfqblSoqr-T1M9sAtFAFym5smmzvtgDY 1693
kithara/model/maxtext/maxtext/MaxText/tests/standalone_dl_ckpt_test.py sha256=Lj_Y6_JvL_mNxUIRlKVRn1Xf7kynfgefKQAp_bF4dbU 3184
kithara/model/maxtext/maxtext/MaxText/tests/tfds_data_processing_test.py sha256=ttU0Zjxd6yirBcHO07OIFVQXrxq_x-gpYGNdEstoNRQ 5026
kithara/model/maxtext/maxtext/MaxText/tests/tokenizer_test.py sha256=WHN8is8PjXNu1kp8wA0_EH2bSNInkCin7LtDyqELmeQ 3428
kithara/model/maxtext/maxtext/MaxText/tests/train_compile_test.py sha256=UKHu5HybttdkGxZGsxjBZx9isyruOcT-CgbLcJ9T7-g 14333
kithara/model/maxtext/maxtext/MaxText/tests/train_gpu_smoke_test.py sha256=gsMI5xhBdcBIoUoKBJpVCOzEJBVO6WTDNKj1H4_j1ao 1374
kithara/model/maxtext/maxtext/MaxText/tests/train_int8_smoke_test.py sha256=hZzdQtLeSU--0gmzGFDrwMwOfUkeeP1o9vFYZt3DQlY 1754
kithara/model/maxtext/maxtext/MaxText/tests/train_smoke_test.py sha256=45iwVwjWtdy-4QY65F95Z6CQKcI_i-GkEQrHCmm7dBY 1705
kithara/model/maxtext/maxtext/MaxText/tests/train_tests.py sha256=wHa3cpUO8iIirLyZVZJg5omoQHRU6fd3j1Y00-Fs6Eo 5988
kithara/model/maxtext/maxtext/MaxText/tests/train_using_ragged_dot_smoke_test.py sha256=Uk3lxeCEK6TACBFXpRelvhjcGebZBkiRQDSKHdDFDwQ 1996
kithara/model/maxtext/maxtext/MaxText/tests/weight_dtypes_test.py sha256=3M5w7qKx2VvSUI23qB5VtbIDt-0km4J3G-y4F9FcS0c 2297
kithara/model/maxtext/maxtext/MaxText/tests/integration_tests/checkpoint_compatibility_test.py sha256=ezb0r0x66svGouBAkPOHZQk0LQNjn4UxkFwT2aqeh5Y 1382
kithara/model/maxtext/maxtext/MaxText/tests/integration_tests/checkpointing_test.py sha256=8IFhY5U9BFQ6ef-4dSlXpcxOlEiZZmlX08IRgXlDlAY 1451
kithara/model/maxtext/maxtext/MaxText/tests/integration_tests/generate_param_only_checkpoint_test.py sha256=PPbEYhR9ePbL3B8SdIer_GQefa9sYZDnznOmjxJzwZs 1658
kithara/model/maxtext/maxtext/MaxText/tests/integration_tests/shmap_collective_matmul_test.py sha256=SIclZh7GsXlmdwrd3WnkzKOXtzkTL6pOHRwjncGZf98 971
kithara/model/maxtext/maxtext/assets/tokenizer sha256=fA9UsRM7T4ntZe2Ditq-OFLxZOZWijjIS1lLMOFUQ6s 809273
kithara/model/maxtext/maxtext/assets/tokenizer.gemma sha256=YaexRzkMZFhdbDVD3W_GNpBsmvOGWlVI8n8xruHUyOI 4241003
kithara/model/maxtext/maxtext/assets/tokenizer.llama2 sha256=nlVq_UQhO2vRviuFDru9mPVIFDeoAhr69Y7n-xgY00c 499723
kithara/model/maxtext/maxtext/assets/tokenizer.mistral-v1 sha256=2t_VbXZnFcYdLveApSWrQ7jm2k3mhlvaPZX9714TQFU 493443
kithara/model/maxtext/maxtext/assets/tokenizer.mistral-v3 sha256=N_ADdN6khljuj10PIYlbm8VcsBA5OWB8gYW_0cbKH4k 587404
kithara/model/maxtext/maxtext/assets/tokenizer_llama3.tiktoken sha256=gunTGXnpKrkpzVREQPEp2ezXl7aeMn-A8X4cUNVVG1U 2183982
kithara/model/maxtext/maxtext/benchmarks/Getting_Started_Benchmarking.md sha256=lHQ-gbSvZc_pLnbWW4z3l6QomUdZwItLbxvlA476aRU 4097
kithara/model/maxtext/maxtext/benchmarks/benchmark_runner.py sha256=2kRKECuqjO3YzFaqyuIgSppYj49dztHAcU49732Veew 9647
kithara/model/maxtext/maxtext/benchmarks/llama2_v6e-256_benchmarks.py sha256=53Y_JZsleMOqNEOhXgWv0MKYdSikY6PQVMQ32r3uDpE 1915
kithara/model/maxtext/maxtext/benchmarks/maxtext_trillium_model_configs.py sha256=ouDjKCWLz181HFAMzqeoPUly3uO64O1BNtENC_c_gdw 34235
kithara/model/maxtext/maxtext/benchmarks/maxtext_v5e_model_configs.py sha256=FtRHlDV1M6AnD5JTGHWFucbLp_vrWJTAiK6JoqChtWg 6422
kithara/model/maxtext/maxtext/benchmarks/maxtext_xpk_runner.py sha256=m3fLxZFM6Gy9yU59hJIG0lABM9hoKj5NWXURhxjZHfY 17812
kithara/model/maxtext/maxtext/benchmarks/xla_flags_library.py sha256=Sfq03ToB1HHa1bykxNMpYv7ZgRKD8v85q7j19TgP6j4 8241
kithara/model/maxtext/maxtext/benchmarks/mmlu/mmlu_categories.py sha256=3aIPxqppbougcRZODz823e-OZDBtbzviGrMzc-RClyA 3570
kithara/model/maxtext/maxtext/benchmarks/mmlu/mmlu_eval.py sha256=DWEM6StQIZsD4mYRa5ADULZAEx0PpSXASXBS41ylhD4 9278
kithara/model/maxtext/maxtext/benchmarks/recipes/__init__.py sha256=F9YEwQBcQp-hO0cPchLhXDaxIUdiZY-Ip_WRaMD_0qY 558
kithara/model/maxtext/maxtext/benchmarks/recipes/args_helper.py sha256=kjJTg7oQd6SnoOPYDjkg2eg3gvdWe_P32LS5KJkiKfM 2805
kithara/model/maxtext/maxtext/benchmarks/recipes/pw_remote_python_recipe.py sha256=-OckXXjr5U8DsXz3D5E17eYP_siFw2YViR5C4_xWmg8 3478
kithara/model/maxtext/maxtext/end_to_end/test_checkpoint_compatibility.sh sha256=vJ_cjv9wXx24dQegUO8o3VoGQyCC-iA_1_SVo2DJFs4 2127
kithara/model/maxtext/maxtext/end_to_end/test_checkpointing.sh sha256=_K0amLu8iGwZZN4R3qUrVRSacxWCfNmSkHO2fgHlCF8 2170
kithara/model/maxtext/maxtext/end_to_end/test_generate_param_only_checkpoint.sh sha256=fa5PMD2__CLJoCGBOlkyIlveGRasGVDNSKzTLJqPWOM 3684
kithara/model/maxtext/maxtext/end_to_end/test_jdi.sh sha256=5iFmnwF3c_rXpFziqK3iWp6ozSZ2eFmyIPnyI82Z_KI 238
kithara/model/maxtext/maxtext/end_to_end/test_profiler.py sha256=gWnLFAsxqcG7AWmwvv3y7s8ba_qnWQmWDoogKyBKbCk 2904
kithara/model/maxtext/maxtext/end_to_end/gpu/a3/test_convergence_125m_params.sh sha256=Gfqmad9OWxZuRZ0hZw7prbsylLq1LFpwtWkQ6r99NUk 3503
kithara/model/maxtext/maxtext/end_to_end/gpu/a3/test_convergence_1b_params.sh sha256=llfFxneVKaWN8QnlV9Kjrw1RKQ8Z_sxpXMPP_n4RZEo 3384
kithara/model/maxtext/maxtext/end_to_end/gpu/a3/test_llama2_7b.sh sha256=04nACKNc45kijDQSzEPujjkZq5Ngh6XDF7NFBPuqnfM 4988
kithara/model/maxtext/maxtext/end_to_end/gpu/mixtral/test_8x7b.sh sha256=fT18SR41qefrWLaNsGvF6RVAlaa21y_KEQmk3WVBHgA 3084
kithara/model/maxtext/maxtext/end_to_end/tpu/eval_assert.py sha256=jwmIs5jk6FLXTNv_eESjG3L88Ay9szaWG8B-QHqxYsI 5123
kithara/model/maxtext/maxtext/end_to_end/tpu/llama_finetuning_test.sh sha256=Watc3eh5Hi-wKirWyyzLIbpnmkzY02FxDLHoBbnZ4uM 1012
kithara/model/maxtext/maxtext/end_to_end/tpu/test_checkpoint_resharding.sh sha256=GZGCPpmKIQTeeYJI-5d755dcjeU5BRMEB6INUlqt09o 991
kithara/model/maxtext/maxtext/end_to_end/tpu/test_convergence_1b_params.sh sha256=-W4WJKMDo49qDgH1G7OZnbYLrORTrPcoFIK_Lp5i0_I 3299
kithara/model/maxtext/maxtext/end_to_end/tpu/test_decode.sh sha256=oram5tiZuNm_it7SyzdHQ2kwAUDxeCP1bde3iXkwwns 2325
kithara/model/maxtext/maxtext/end_to_end/tpu/test_decode_load_quantized_ckpt.sh sha256=tzjMf9hGUeUuDopUGbIvlDoQhzoNCBXFzxBTaGQtqTM 2012
kithara/model/maxtext/maxtext/end_to_end/tpu/test_decode_save_quantized_ckpt.sh sha256=3R8izhvBvVMCjCN0c4wvyQL26HwxKLk8i40yxBlUZcE 2333
kithara/model/maxtext/maxtext/end_to_end/tpu/test_determinism.sh sha256=lQ-kWCHAw1LkpSb_j2-yZXtcd8LL80tvB8Nz1HdSyWg 1169
kithara/model/maxtext/maxtext/end_to_end/tpu/test_dpo.sh sha256=UaUmIwUusCPollFrl_KKpdwOeo82O-Rf8wmEmNMmlbM 1887
kithara/model/maxtext/maxtext/end_to_end/tpu/test_gpt3.sh sha256=63SwMrRxcdsQUkF0fHUQqk-BC8abbMiOHhZ19Cap_qA 758
kithara/model/maxtext/maxtext/end_to_end/tpu/test_tflops.sh sha256=gi3J-2H0sIWy0IyDCn_Ks51IFroMfZia4V7OA3ReoWU 606
kithara/model/maxtext/maxtext/end_to_end/tpu/test_tflops_16b_params.sh sha256=ow4uumSxstjjmr-Hta6fN4rzzAtO64BGVhi37YQr_0c 1789
kithara/model/maxtext/maxtext/end_to_end/tpu/test_tflops_32b_params.sh sha256=WGYKK9AWqQSbOXEIfbegoM5C7uZjhOVmEJjNakzd-gA 1789
kithara/model/maxtext/maxtext/end_to_end/tpu/test_tflops_64b_params.sh sha256=jRyD3hpfDvXvHAdECoonY9WWutjP_4V5fcgYjBXBNI0 1789
kithara/model/maxtext/maxtext/end_to_end/tpu/test_vocab_creation.sh sha256=P1DRr7_g4CVOi0rOJVMCMhzfuap785iXKhis6ntsP2U 418
kithara/model/maxtext/maxtext/end_to_end/tpu/gemma/Run_Gemma.md sha256=sIky5M_FBHkQtJb00NeZg0bl6v8jId0CN7dGl4rlm2M 1945
kithara/model/maxtext/maxtext/end_to_end/tpu/gemma/2b/test_gemma.sh sha256=Rcyf0Q-7OY6PNA3PJXfcA7W1qIDsMBw64onVowMwq1s 7650
kithara/model/maxtext/maxtext/end_to_end/tpu/gemma/7b/1_test_gemma.sh sha256=FCLzLLTsgCqi-aoE-zXLQVGBGh8HNOLAjbP17Z7V9FQ 3009
kithara/model/maxtext/maxtext/end_to_end/tpu/gemma/7b/2_test_gemma.sh sha256=O-WPhEEs98tVCT2u3h5F7ZfUVO2ACvEftmkpfNZpvP4 6967
kithara/model/maxtext/maxtext/end_to_end/tpu/gemma2/27b/1_test_gemma.sh sha256=DWQKrqjFQOko3umYNAHp_x3VpVmMT6jAqfP2Ho2Etbk 3125
kithara/model/maxtext/maxtext/end_to_end/tpu/gemma2/27b/2_test_gemma.sh sha256=_mnnw5D_TOsYIoT4fUGVqtMcZfB02N5S7HxS8kldEAc 4219
kithara/model/maxtext/maxtext/end_to_end/tpu/gemma2/2b/test_gemma2.sh sha256=bOToCaPvMcmm_gbpK2-5XX09ES9z1qXWqS5BbN4hxDU 7132
kithara/model/maxtext/maxtext/end_to_end/tpu/gemma2/9b/1_test_gemma.sh sha256=TSt2oUVMy6RXrlXWdr3HoyYCY3OcEdbzGzVStTrHMZY 3120
kithara/model/maxtext/maxtext/end_to_end/tpu/gemma2/9b/2_test_gemma.sh sha256=XjWjQkWSIkv-E0-EDvswQo7gYi8XijBDciA5A2Wx5Ow 4204
kithara/model/maxtext/maxtext/end_to_end/tpu/llama2/13b/1_test_llama2_13b.sh sha256=Z3JRi4DKl8SOyF4Db-9Fa9AMItAaGJ_NygD64aUfpFE 3652
kithara/model/maxtext/maxtext/end_to_end/tpu/llama2/13b/2_test_llama2_13b.sh sha256=c3wxuLFbhF-qJUEpOIZaFgW3Cowll3LMgUefZ1TRfL0 6286
kithara/model/maxtext/maxtext/end_to_end/tpu/llama2/70b/1_test_llama2_70b.sh sha256=tZ2O8kvmiTS0cMYGf0PRFE78DvOFpng29fwlk9vlS28 3652
kithara/model/maxtext/maxtext/end_to_end/tpu/llama2/70b/2_test_llama2_70b.sh sha256=1vUNaPv7v9bIHBLuCVQ7M6Z7DSAg6LParIDD53E2q_M 6984
kithara/model/maxtext/maxtext/end_to_end/tpu/llama2/7b/test_llama2_7b.sh sha256=GnG3ct82iCaO1jMyjCMopcTeV8j5FxBeVxGcp6Pds1A 9121
kithara/model/maxtext/maxtext/end_to_end/tpu/llama3/70b/1_test_llama3_70b.sh sha256=2xvYqaBI8xBIODL849FTFf5Fx_yIxsdmxwdRAieJOFM 3673
kithara/model/maxtext/maxtext/end_to_end/tpu/llama3/70b/2_test_llama3_70b.sh sha256=Eywxbsgu3Y_PaGMN4VHcviW-QwHj9jrTKNqj3qYrkyA 6713
kithara/model/maxtext/maxtext/end_to_end/tpu/llama3/8b/1_test_llama3_8b.sh sha256=TzNKwrMbfFaZ4t5MKnKCBTh9QEC3fHrW8NPEwrlm9CA 3643
kithara/model/maxtext/maxtext/end_to_end/tpu/llama3/8b/2_test_llama3_8b.sh sha256=2krgquni160gUfbADwkpOPbseolEJhogZGtIRU_k2mQ 6704
kithara/model/maxtext/maxtext/end_to_end/tpu/llama3.1/405b/2_test_llama3.1_405b.sh sha256=ttRFEeBLHa_4detaZ5IROwwPxO6EebhUKpiSn_Ba304 4465
kithara/model/maxtext/maxtext/end_to_end/tpu/llama3.1/70b/1_test_llama3.1_70b.sh sha256=eVgkQpxygL8x1Nmya19mkabC92xTdcDRKEaPdkRfSGo 3665
kithara/model/maxtext/maxtext/end_to_end/tpu/llama3.1/70b/2_test_llama3.1_70b.sh sha256=cBMowgh31qDj0l_9yeuLLqsgboaceGktP77g-G28ZpI 5024
kithara/model/maxtext/maxtext/end_to_end/tpu/llama3.1/8b/1_test_llama3.1_8b.sh sha256=_JRz4jIpvLJ54MlAy41K-NrgOmpshgA2eQ9vmoLkm90 3672
kithara/model/maxtext/maxtext/end_to_end/tpu/llama3.1/8b/2_test_llama3.1_8b.sh sha256=sYTHkyBXIRZi-QRdFSFLjAkqBw1OFfYkHpgxMqbhyj4 6054
kithara/model/maxtext/maxtext/end_to_end/tpu/mistral/7b/test_mistral-7b.sh sha256=psrSh0zPz-2MkUAOljxmr3e-H_wT36qvMtHG-5oQMC0 3177
kithara/model/maxtext/maxtext/end_to_end/tpu/mixtral/Run_Mixtral.md sha256=th2AQcQ4qf8-UtTL5k9JEmil0e1t8STe6znqwlwnB4k 1748
kithara/model/maxtext/maxtext/end_to_end/tpu/mixtral/8x22b/1_test_mixtral.sh sha256=C9NYx9Vrk8MgSSFTOL3hYIP2hze1ku6cmG9y0U6tejw 2006
kithara/model/maxtext/maxtext/end_to_end/tpu/mixtral/8x22b/2_test_mixtral.sh sha256=wDOvZnARq46O-nGdeC1KocGQRNQjOYJW4pW9pgbI_VU 2970
kithara/model/maxtext/maxtext/end_to_end/tpu/mixtral/8x7b/1_test_mixtral.sh sha256=NgJ2cHSaU0Hq8acI4J57IZ__3tkvkbE9CqOUiuPQ1IA 2256
kithara/model/maxtext/maxtext/end_to_end/tpu/mixtral/8x7b/2_test_mixtral.sh sha256=1h2OCc4GwoE9ohnnMKPRoOO_smgOmQkGyqCrcLTRB0k 5317
kithara/model/maxtext/maxtext/getting_started/Data_Input_Perf.md sha256=xWYRUNGWfABG2_W5iupLvwgVWzlO446YxtgiBrfWeBo 3467
kithara/model/maxtext/maxtext/getting_started/Data_Input_Pipeline.md sha256=HqCB1qbiNmcDme4J6odW9e1WPdfS_bwsfD8HeMV8Nns 12361
kithara/model/maxtext/maxtext/getting_started/First_run.md sha256=937sm3S0aColaFWsCO99jSrRWB4g6B_fozkQnPKn_fQ 4284
kithara/model/maxtext/maxtext/getting_started/GCP_Workload_Observability.md sha256=r4Q2_SkxvSYl2o36Kvwm_jx-UvoQXSgORqOTtun_ztQ 1621
kithara/model/maxtext/maxtext/getting_started/Run_Llama2.md sha256=NQxapIaU-LrA6uNYhYjCxUVbNo3B7alxkuCbcbCNpCo 1697
kithara/model/maxtext/maxtext/getting_started/Run_MaxText_via_multihost_job.md sha256=qet5sEymWL1qyMbGiSlKBg8NFVOmtIcJSjiaLNVrkC8 3732
kithara/model/maxtext/maxtext/getting_started/Run_MaxText_via_multihost_runner.md sha256=VlrW5RPpa3DYRsc3GoxBJxiFOhv5cAc9pyHpKBXZ2HQ 5154
kithara/model/maxtext/maxtext/getting_started/Run_MaxText_via_xpk.md sha256=KQoXT1VyxZsOPrlDXIQ_v7iNOw-GJxfYC0bQAFpP-Tc 5710
kithara/model/maxtext/maxtext/getting_started/Use_Vertex_AI_Tensorboard.md sha256=zcfFtZTA-kHODYwnIYhMhkkomBIczWCo8RjdLaef2iE 4771
kithara/model/maxtext/maxtext/inference/maxengine_server/Dockerfile sha256=J2Q4N3dd6pYqKxNLpcg1LcgyHx8xdH5_HfTSXtx3Etc 1442
kithara/model/maxtext/maxtext/inference/maxengine_server/README.md sha256=NoAjdcjx1RM9EctXOtfqwW5LfVWqeD1UQByMnBPr3js 676
kithara/model/maxtext/maxtext/inference/maxengine_server/maxengine_server_entrypoint.sh sha256=nsqszieIVMJG88VuQ0cZvnDe_yjwbmBKOgBTV0l2v2o 665
kithara/model/maxtext/maxtext/pedagogical_examples/non_spmd.py sha256=UCVmUvfJK1UHi4UqjGYk-UPFo7faMQm0sys0JNFNiQY 1813
kithara/model/maxtext/maxtext/pedagogical_examples/shardings.py sha256=9kFQ_aY4pXtZXJV51FZ3HEYAdvwqnG8YhjY22AB-1lQ 8949
kithara/model/maxtext/maxtext/pedagogical_examples/shmap_collective_matmul.py sha256=h4-6NMnIZmMh-BQsUKL8dIqkmnCpEfFM4MA5269nmlc 6916
kithara/trainer/__init__.py sha256=RF2u72vRBGRCtsmMuKkVw49zI2bcUMs6kp3kagngr3Q 37
kithara/trainer/trainer.py sha256=uhkgi76Nawrhz6G9zf3IYeD3g71p1LBFNGPQJ-8RC-4 28288
kithara/utils/gcs_utils.py sha256=c-nvO5HjwGrIYUI17iBQcUsdturMx5hVHOybZfNaSo8 3524
kithara/utils/logging_utils.py sha256=x-FnwXcrmEaUzkuJvsrWizjgVIe3qUIfMElAG-mrOA4 1244
kithara/utils/safetensor_utils.py sha256=Xr73PzsLj3jEfLxNC2FA5KRv3cN0AloEyxY-2-FNmJI 6626
kithara/utils/torch_utils.py sha256=LsT3fqwxPXC4T5OSYDygSVO6BsRku1g8qTOVS8sq0sE 1143
kithara/utils/tree_utils.py sha256=iwDqkH1db9NGI-hBXgXgKsveHJgWsyIkbtn4_Mppe3M 1553
kithara-0.0.8.data/data/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
kithara-0.0.8.data/data/main.py sha256=QaTf7Tp7Uhn8NiAseV8m9UaBMBueuPMQInGe7OQD1cU 1442
kithara-0.0.8.data/data/multihost.py sha256=zZD7tRXIuQxaaUO6HKN6hAkHgoZb1HjziOhxPYaaAgg 5132
kithara-0.0.8.dist-info/entry_points.txt sha256=vvuf3FHFtyovdMKvsE1zxw6WuLC4hBTC2Wh32QuLm4s 90
kithara-0.0.8.dist-info/LICENSE sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0 11357
kithara-0.0.8.dist-info/WHEEL sha256=CpUCUxeHQbRN5UGRQHYRJorO5Af-Qy_fHMctcQ8DSGI 82
kithara-0.0.8.dist-info/METADATA sha256=j3epQGmaKwKl_4_0U35xmoDyIXVK_uzAQzq3dlcH6aM 4969
kithara-0.0.8.dist-info/RECORD

entry_points.txt

kithara = cli.main:main
multihost = cli.multihost:main