AI-Hypercomputer
diff --git a/‎google3/third_party/py/jetstream/.github/CODEOWNERS renamed to ‎.github/CODEOWNERS b/‎google3/third_party/py/jetstream/.github/CODEOWNERS renamed to ‎.github/CODEOWNERS
diff --git a/‎google3/third_party/py/jetstream/.github/workflows/e2e_tests.yaml renamed to ‎.github/workflows/e2e_tests.yaml b/‎google3/third_party/py/jetstream/.github/workflows/e2e_tests.yaml renamed to ‎.github/workflows/e2e_tests.yaml
diff --git a/‎google3/third_party/py/jetstream/.github/workflows/release.yaml renamed to ‎.github/workflows/release.yaml b/‎google3/third_party/py/jetstream/.github/workflows/release.yaml renamed to ‎.github/workflows/release.yaml
diff --git a/‎google3/third_party/py/jetstream/.github/workflows/scripts/create_release.js renamed to ‎.github/workflows/scripts/create_release.js b/‎google3/third_party/py/jetstream/.github/workflows/scripts/create_release.js renamed to ‎.github/workflows/scripts/create_release.js
diff --git a/‎google3/third_party/py/jetstream/.github/workflows/unit_tests.yaml renamed to ‎.github/workflows/unit_tests.yaml b/‎google3/third_party/py/jetstream/.github/workflows/unit_tests.yaml renamed to ‎.github/workflows/unit_tests.yaml
diff --git a/‎google3/third_party/py/jetstream/.gitignore renamed to ‎.gitignore b/‎google3/third_party/py/jetstream/.gitignore renamed to ‎.gitignore
diff --git a/‎google3/third_party/py/jetstream/AUTHORS renamed to ‎AUTHORS b/‎google3/third_party/py/jetstream/AUTHORS renamed to ‎AUTHORS
diff --git a/‎google3/third_party/py/jetstream/CONTRIBUTING.md renamed to ‎CONTRIBUTING.md b/‎google3/third_party/py/jetstream/CONTRIBUTING.md renamed to ‎CONTRIBUTING.md
diff --git a/‎google3/third_party/py/jetstream/LICENSE renamed to ‎LICENSE b/‎google3/third_party/py/jetstream/LICENSE renamed to ‎LICENSE
diff --git a/‎google3/third_party/py/jetstream/MANIFEST.in renamed to ‎MANIFEST.in b/‎google3/third_party/py/jetstream/MANIFEST.in renamed to ‎MANIFEST.in
diff --git a/‎google3/third_party/py/jetstream/Makefile renamed to ‎Makefile b/‎google3/third_party/py/jetstream/Makefile renamed to ‎Makefile
diff --git a/‎google3/third_party/py/jetstream/README.md renamed to ‎README.md b/‎google3/third_party/py/jetstream/README.md renamed to ‎README.md
diff --git a/‎google3/third_party/py/jetstream/benchmarks/README.md renamed to ‎benchmarks/README.md b/‎google3/third_party/py/jetstream/benchmarks/README.md renamed to ‎benchmarks/README.md
diff --git a/‎google3/third_party/py/jetstream/benchmarks/__init__.py renamed to ‎benchmarks/__init__.py b/‎google3/third_party/py/jetstream/benchmarks/__init__.py renamed to ‎benchmarks/__init__.py
diff --git a/‎google3/third_party/py/jetstream/benchmarks/benchmark_serving.py renamed to ‎benchmarks/benchmark_serving.py b/‎google3/third_party/py/jetstream/benchmarks/benchmark_serving.py renamed to ‎benchmarks/benchmark_serving.py
diff --git a/‎google3/third_party/py/jetstream/benchmarks/eval_accuracy.py renamed to ‎benchmarks/eval_accuracy.py
+1 b/‎google3/third_party/py/jetstream/benchmarks/eval_accuracy.py renamed to ‎benchmarks/eval_accuracy.py
+1
diff --git a/‎google3/third_party/py/jetstream/benchmarks/open_orca_gpt4_tokenized_llama.calibration_1000.pkl renamed to ‎benchmarks/open_orca_gpt4_tokenized_llama.calibration_1000.pkl b/‎google3/third_party/py/jetstream/benchmarks/open_orca_gpt4_tokenized_llama.calibration_1000.pkl renamed to ‎benchmarks/open_orca_gpt4_tokenized_llama.calibration_1000.pkl
diff --git a/‎google3/third_party/py/jetstream/benchmarks/requirements.in renamed to ‎benchmarks/requirements.in b/‎google3/third_party/py/jetstream/benchmarks/requirements.in renamed to ‎benchmarks/requirements.in
diff --git a/‎google3/third_party/py/jetstream/docs/observability-prometheus-metrics-in-jetstream-server.md renamed to ‎docs/observability-prometheus-metrics-in-jetstream-server.md
+1-1 b/‎google3/third_party/py/jetstream/docs/observability-prometheus-metrics-in-jetstream-server.md renamed to ‎docs/observability-prometheus-metrics-in-jetstream-server.md
+1-1
diff --git a/‎google3/third_party/py/jetstream/docs/online-inference-with-maxtext-engine.md renamed to ‎docs/online-inference-with-maxtext-engine.md b/‎google3/third_party/py/jetstream/docs/online-inference-with-maxtext-engine.md renamed to ‎docs/online-inference-with-maxtext-engine.md
diff --git a/‎google3/third_party/py/jetstream/docs/profiling-with-jax-profiler-and-tensorboard.md renamed to ‎docs/profiling-with-jax-profiler-and-tensorboard.md b/‎google3/third_party/py/jetstream/docs/profiling-with-jax-profiler-and-tensorboard.md renamed to ‎docs/profiling-with-jax-profiler-and-tensorboard.md
diff --git a/‎google3/third_party/py/jetstream/__init__.py renamed to ‎jetstream/__init__.py b/‎google3/third_party/py/jetstream/__init__.py renamed to ‎jetstream/__init__.py
diff --git a/‎google3/third_party/py/jetstream/core/README.md renamed to ‎jetstream/core/README.md b/‎google3/third_party/py/jetstream/core/README.md renamed to ‎jetstream/core/README.md
diff --git a/‎google3/third_party/py/jetstream/core/__init__.py renamed to ‎jetstream/core/__init__.py b/‎google3/third_party/py/jetstream/core/__init__.py renamed to ‎jetstream/core/__init__.py
diff --git a/‎google3/third_party/py/jetstream/core/config_lib.py renamed to ‎jetstream/core/config_lib.py b/‎google3/third_party/py/jetstream/core/config_lib.py renamed to ‎jetstream/core/config_lib.py
diff --git a/‎google3/third_party/py/jetstream/core/implementations/__init__.py renamed to ‎jetstream/core/implementations/__init__.py b/‎google3/third_party/py/jetstream/core/implementations/__init__.py renamed to ‎jetstream/core/implementations/__init__.py
diff --git a/‎google3/third_party/py/jetstream/core/implementations/mock/README.md renamed to ‎jetstream/core/implementations/mock/README.md b/‎google3/third_party/py/jetstream/core/implementations/mock/README.md renamed to ‎jetstream/core/implementations/mock/README.md
diff --git a/‎google3/third_party/py/jetstream/core/implementations/mock/__init__.py renamed to ‎jetstream/core/implementations/mock/__init__.py b/‎google3/third_party/py/jetstream/core/implementations/mock/__init__.py renamed to ‎jetstream/core/implementations/mock/__init__.py
diff --git a/‎google3/third_party/py/jetstream/core/implementations/mock/config.py renamed to ‎jetstream/core/implementations/mock/config.py b/‎google3/third_party/py/jetstream/core/implementations/mock/config.py renamed to ‎jetstream/core/implementations/mock/config.py
diff --git a/‎google3/third_party/py/jetstream/core/implementations/mock/server.py renamed to ‎jetstream/core/implementations/mock/server.py b/‎google3/third_party/py/jetstream/core/implementations/mock/server.py renamed to ‎jetstream/core/implementations/mock/server.py
diff --git a/‎google3/third_party/py/jetstream/core/metrics/__init__.py renamed to ‎jetstream/core/metrics/__init__.py b/‎google3/third_party/py/jetstream/core/metrics/__init__.py renamed to ‎jetstream/core/metrics/__init__.py
diff --git a/‎google3/third_party/py/jetstream/core/metrics/prometheus.py renamed to ‎jetstream/core/metrics/prometheus.py b/‎google3/third_party/py/jetstream/core/metrics/prometheus.py renamed to ‎jetstream/core/metrics/prometheus.py
diff --git a/‎google3/third_party/py/jetstream/core/orchestrator.py renamed to ‎jetstream/core/orchestrator.py b/‎google3/third_party/py/jetstream/core/orchestrator.py renamed to ‎jetstream/core/orchestrator.py
diff --git a/‎google3/third_party/py/jetstream/core/proto/__init__.py renamed to ‎jetstream/core/proto/__init__.py b/‎google3/third_party/py/jetstream/core/proto/__init__.py renamed to ‎jetstream/core/proto/__init__.py
diff --git a/‎google3/third_party/py/jetstream/core/proto/jetstream.proto renamed to ‎jetstream/core/proto/jetstream.proto b/‎google3/third_party/py/jetstream/core/proto/jetstream.proto renamed to ‎jetstream/core/proto/jetstream.proto
diff --git a/‎google3/third_party/py/jetstream/core/proto/jetstream_pb2.py renamed to ‎jetstream/core/proto/jetstream_pb2.py b/‎google3/third_party/py/jetstream/core/proto/jetstream_pb2.py renamed to ‎jetstream/core/proto/jetstream_pb2.py
diff --git a/‎google3/third_party/py/jetstream/core/proto/jetstream_pb2_grpc.py renamed to ‎jetstream/core/proto/jetstream_pb2_grpc.py b/‎google3/third_party/py/jetstream/core/proto/jetstream_pb2_grpc.py renamed to ‎jetstream/core/proto/jetstream_pb2_grpc.py
diff --git a/‎google3/third_party/py/jetstream/core/server_lib.py renamed to ‎jetstream/core/server_lib.py b/‎google3/third_party/py/jetstream/core/server_lib.py renamed to ‎jetstream/core/server_lib.py
diff --git a/‎google3/third_party/py/jetstream/core/utils/__init__.py renamed to ‎jetstream/core/utils/__init__.py b/‎google3/third_party/py/jetstream/core/utils/__init__.py renamed to ‎jetstream/core/utils/__init__.py
diff --git a/‎google3/third_party/py/jetstream/core/utils/async_multifuture.py renamed to ‎jetstream/core/utils/async_multifuture.py b/‎google3/third_party/py/jetstream/core/utils/async_multifuture.py renamed to ‎jetstream/core/utils/async_multifuture.py
diff --git a/‎google3/third_party/py/jetstream/core/utils/proxy_util.py renamed to ‎jetstream/core/utils/proxy_util.py b/‎google3/third_party/py/jetstream/core/utils/proxy_util.py renamed to ‎jetstream/core/utils/proxy_util.py
diff --git a/‎google3/third_party/py/jetstream/core/utils/return_sample.py renamed to ‎jetstream/core/utils/return_sample.py b/‎google3/third_party/py/jetstream/core/utils/return_sample.py renamed to ‎jetstream/core/utils/return_sample.py
diff --git a/‎google3/third_party/py/jetstream/engine/README.md renamed to ‎jetstream/engine/README.md b/‎google3/third_party/py/jetstream/engine/README.md renamed to ‎jetstream/engine/README.md
diff --git a/‎google3/third_party/py/jetstream/engine/__init__.py renamed to ‎jetstream/engine/__init__.py b/‎google3/third_party/py/jetstream/engine/__init__.py renamed to ‎jetstream/engine/__init__.py
diff --git a/‎google3/third_party/py/jetstream/engine/engine_api.py renamed to ‎jetstream/engine/engine_api.py b/‎google3/third_party/py/jetstream/engine/engine_api.py renamed to ‎jetstream/engine/engine_api.py
diff --git a/‎google3/third_party/py/jetstream/engine/mock_engine.py renamed to ‎jetstream/engine/mock_engine.py b/‎google3/third_party/py/jetstream/engine/mock_engine.py renamed to ‎jetstream/engine/mock_engine.py
diff --git a/‎google3/third_party/py/jetstream/engine/mock_utils.py renamed to ‎jetstream/engine/mock_utils.py b/‎google3/third_party/py/jetstream/engine/mock_utils.py renamed to ‎jetstream/engine/mock_utils.py
diff --git a/‎google3/third_party/py/jetstream/engine/sampling_utils.py renamed to ‎jetstream/engine/sampling_utils.py b/‎google3/third_party/py/jetstream/engine/sampling_utils.py renamed to ‎jetstream/engine/sampling_utils.py
diff --git a/‎google3/third_party/py/jetstream/engine/token_utils.py renamed to ‎jetstream/engine/token_utils.py b/‎google3/third_party/py/jetstream/engine/token_utils.py renamed to ‎jetstream/engine/token_utils.py
diff --git a/‎google3/third_party/py/jetstream/engine/tokenizer.proto renamed to ‎jetstream/engine/tokenizer.proto b/‎google3/third_party/py/jetstream/engine/tokenizer.proto renamed to ‎jetstream/engine/tokenizer.proto
diff --git a/‎google3/third_party/py/jetstream/engine/tokenizer_api.py renamed to ‎jetstream/engine/tokenizer_api.py b/‎google3/third_party/py/jetstream/engine/tokenizer_api.py renamed to ‎jetstream/engine/tokenizer_api.py
diff --git a/‎google3/third_party/py/jetstream/engine/tokenizer_pb2.py renamed to ‎jetstream/engine/tokenizer_pb2.py b/‎google3/third_party/py/jetstream/engine/tokenizer_pb2.py renamed to ‎jetstream/engine/tokenizer_pb2.py
diff --git a/‎google3/third_party/py/jetstream/engine/tokenizer_pb2_grpc.py renamed to ‎jetstream/engine/tokenizer_pb2_grpc.py b/‎google3/third_party/py/jetstream/engine/tokenizer_pb2_grpc.py renamed to ‎jetstream/engine/tokenizer_pb2_grpc.py
diff --git a/‎google3/third_party/py/jetstream/engine/warmup_utils.py renamed to ‎jetstream/engine/warmup_utils.py b/‎google3/third_party/py/jetstream/engine/warmup_utils.py renamed to ‎jetstream/engine/warmup_utils.py
diff --git a/‎google3/third_party/py/jetstream/entrypoints/__init__.py renamed to ‎jetstream/entrypoints/__init__.py b/‎google3/third_party/py/jetstream/entrypoints/__init__.py renamed to ‎jetstream/entrypoints/__init__.py
diff --git a/‎google3/third_party/py/jetstream/entrypoints/config.py renamed to ‎jetstream/entrypoints/config.py b/‎google3/third_party/py/jetstream/entrypoints/config.py renamed to ‎jetstream/entrypoints/config.py
diff --git a/‎google3/third_party/py/jetstream/entrypoints/http/__init__.py renamed to ‎jetstream/entrypoints/http/__init__.py b/‎google3/third_party/py/jetstream/entrypoints/http/__init__.py renamed to ‎jetstream/entrypoints/http/__init__.py
diff --git a/‎google3/third_party/py/jetstream/entrypoints/http/api_server.py renamed to ‎jetstream/entrypoints/http/api_server.py b/‎google3/third_party/py/jetstream/entrypoints/http/api_server.py renamed to ‎jetstream/entrypoints/http/api_server.py
diff --git a/‎google3/third_party/py/jetstream/entrypoints/http/protocol.py renamed to ‎jetstream/entrypoints/http/protocol.py b/‎google3/third_party/py/jetstream/entrypoints/http/protocol.py renamed to ‎jetstream/entrypoints/http/protocol.py
diff --git a/‎google3/third_party/py/jetstream/entrypoints/http/utils.py renamed to ‎jetstream/entrypoints/http/utils.py b/‎google3/third_party/py/jetstream/entrypoints/http/utils.py renamed to ‎jetstream/entrypoints/http/utils.py
diff --git a/‎google3/third_party/py/jetstream/external_tokenizers/__init__.py renamed to ‎jetstream/external_tokenizers/__init__.py b/‎google3/third_party/py/jetstream/external_tokenizers/__init__.py renamed to ‎jetstream/external_tokenizers/__init__.py
diff --git a/‎google3/third_party/py/jetstream/external_tokenizers/llama3/__init__.py renamed to ‎jetstream/external_tokenizers/llama3/__init__.py b/‎google3/third_party/py/jetstream/external_tokenizers/llama3/__init__.py renamed to ‎jetstream/external_tokenizers/llama3/__init__.py
diff --git a/‎google3/third_party/py/jetstream/external_tokenizers/llama3/llama3_tokenizer.py renamed to ‎jetstream/external_tokenizers/llama3/llama3_tokenizer.py b/‎google3/third_party/py/jetstream/external_tokenizers/llama3/llama3_tokenizer.py renamed to ‎jetstream/external_tokenizers/llama3/llama3_tokenizer.py
diff --git a/‎google3/third_party/py/jetstream/tests/__init__.py renamed to ‎jetstream/tests/__init__.py b/‎google3/third_party/py/jetstream/tests/__init__.py renamed to ‎jetstream/tests/__init__.py
diff --git a/‎google3/third_party/py/jetstream/tests/core/__init__.py renamed to ‎jetstream/tests/core/__init__.py b/‎google3/third_party/py/jetstream/tests/core/__init__.py renamed to ‎jetstream/tests/core/__init__.py
diff --git a/‎google3/third_party/py/jetstream/tests/core/test_config_lib.py renamed to ‎jetstream/tests/core/test_config_lib.py b/‎google3/third_party/py/jetstream/tests/core/test_config_lib.py renamed to ‎jetstream/tests/core/test_config_lib.py
diff --git a/‎google3/third_party/py/jetstream/tests/core/test_orchestrator.py renamed to ‎jetstream/tests/core/test_orchestrator.py b/‎google3/third_party/py/jetstream/tests/core/test_orchestrator.py renamed to ‎jetstream/tests/core/test_orchestrator.py
diff --git a/‎google3/third_party/py/jetstream/tests/core/test_server.py renamed to ‎jetstream/tests/core/test_server.py b/‎google3/third_party/py/jetstream/tests/core/test_server.py renamed to ‎jetstream/tests/core/test_server.py
diff --git a/‎google3/third_party/py/jetstream/tests/engine/__init__.py renamed to ‎jetstream/tests/engine/__init__.py b/‎google3/third_party/py/jetstream/tests/engine/__init__.py renamed to ‎jetstream/tests/engine/__init__.py
diff --git a/‎google3/third_party/py/jetstream/tests/engine/external_tokenizers/llama2/tokenizer.model renamed to ‎jetstream/tests/engine/external_tokenizers/llama2/tokenizer.model b/‎google3/third_party/py/jetstream/tests/engine/external_tokenizers/llama2/tokenizer.model renamed to ‎jetstream/tests/engine/external_tokenizers/llama2/tokenizer.model
diff --git a/‎google3/third_party/py/jetstream/tests/engine/external_tokenizers/llama3/tokenizer.model renamed to ‎jetstream/tests/engine/external_tokenizers/llama3/tokenizer.model b/‎google3/third_party/py/jetstream/tests/engine/external_tokenizers/llama3/tokenizer.model renamed to ‎jetstream/tests/engine/external_tokenizers/llama3/tokenizer.model
diff --git a/‎google3/third_party/py/jetstream/tests/engine/test_mock_engine.py renamed to ‎jetstream/tests/engine/test_mock_engine.py b/‎google3/third_party/py/jetstream/tests/engine/test_mock_engine.py renamed to ‎jetstream/tests/engine/test_mock_engine.py
diff --git a/‎google3/third_party/py/jetstream/tests/engine/test_sampling_utils.py renamed to ‎jetstream/tests/engine/test_sampling_utils.py b/‎google3/third_party/py/jetstream/tests/engine/test_sampling_utils.py renamed to ‎jetstream/tests/engine/test_sampling_utils.py
diff --git a/‎google3/third_party/py/jetstream/tests/engine/test_token_utils.py renamed to ‎jetstream/tests/engine/test_token_utils.py b/‎google3/third_party/py/jetstream/tests/engine/test_token_utils.py renamed to ‎jetstream/tests/engine/test_token_utils.py
diff --git a/‎google3/third_party/py/jetstream/tests/engine/test_utils.py renamed to ‎jetstream/tests/engine/test_utils.py b/‎google3/third_party/py/jetstream/tests/engine/test_utils.py renamed to ‎jetstream/tests/engine/test_utils.py
diff --git a/‎google3/third_party/py/jetstream/tests/entrypoints/__init__.py renamed to ‎jetstream/tests/entrypoints/__init__.py b/‎google3/third_party/py/jetstream/tests/entrypoints/__init__.py renamed to ‎jetstream/tests/entrypoints/__init__.py
diff --git a/‎google3/third_party/py/jetstream/tests/entrypoints/http/__init__.py renamed to ‎jetstream/tests/entrypoints/http/__init__.py b/‎google3/third_party/py/jetstream/tests/entrypoints/http/__init__.py renamed to ‎jetstream/tests/entrypoints/http/__init__.py
diff --git a/‎google3/third_party/py/jetstream/tests/entrypoints/http/test_api_server.py renamed to ‎jetstream/tests/entrypoints/http/test_api_server.py b/‎google3/third_party/py/jetstream/tests/entrypoints/http/test_api_server.py renamed to ‎jetstream/tests/entrypoints/http/test_api_server.py
diff --git a/‎google3/third_party/py/jetstream/tools/load_tester.py renamed to ‎jetstream/tools/load_tester.py b/‎google3/third_party/py/jetstream/tools/load_tester.py renamed to ‎jetstream/tools/load_tester.py
diff --git a/‎google3/third_party/py/jetstream/tools/maxtext/model_ckpt_conversion.sh renamed to ‎jetstream/tools/maxtext/model_ckpt_conversion.sh b/‎google3/third_party/py/jetstream/tools/maxtext/model_ckpt_conversion.sh renamed to ‎jetstream/tools/maxtext/model_ckpt_conversion.sh
diff --git a/‎google3/third_party/py/jetstream/tools/maxtext/model_ckpt_finetune_with_aqt.sh renamed to ‎jetstream/tools/maxtext/model_ckpt_finetune_with_aqt.sh b/‎google3/third_party/py/jetstream/tools/maxtext/model_ckpt_finetune_with_aqt.sh renamed to ‎jetstream/tools/maxtext/model_ckpt_finetune_with_aqt.sh
diff --git a/‎google3/third_party/py/jetstream/tools/proxy_dev/base.Dockerfile renamed to ‎jetstream/tools/proxy_dev/base.Dockerfile b/‎google3/third_party/py/jetstream/tools/proxy_dev/base.Dockerfile renamed to ‎jetstream/tools/proxy_dev/base.Dockerfile
diff --git a/‎google3/third_party/py/jetstream/tools/proxy_dev/dev.Dockerfile renamed to ‎jetstream/tools/proxy_dev/dev.Dockerfile b/‎google3/third_party/py/jetstream/tools/proxy_dev/dev.Dockerfile renamed to ‎jetstream/tools/proxy_dev/dev.Dockerfile
diff --git a/‎google3/third_party/py/jetstream/tools/requester.py renamed to ‎jetstream/tools/requester.py b/‎google3/third_party/py/jetstream/tools/requester.py renamed to ‎jetstream/tools/requester.py
diff --git a/‎google3/third_party/py/jetstream/license_preamble.txt renamed to ‎license_preamble.txt b/‎google3/third_party/py/jetstream/license_preamble.txt renamed to ‎license_preamble.txt
diff --git a/‎google3/third_party/py/jetstream/pylintrc renamed to ‎pylintrc b/‎google3/third_party/py/jetstream/pylintrc renamed to ‎pylintrc
diff --git a/‎google3/third_party/py/jetstream/requirements.txt renamed to ‎requirements.txt b/‎google3/third_party/py/jetstream/requirements.txt renamed to ‎requirements.txt
diff --git a/‎google3/third_party/py/jetstream/setup.py renamed to ‎setup.py b/‎google3/third_party/py/jetstream/setup.py renamed to ‎setup.py
@@ -64,6 +64,7 @@ def main(args):
 
   eval_accuracy(request_outputs_dict)
 
+
 if __name__ == "__main__":
   parser = argparse.ArgumentParser()
   parser.add_argument(