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