├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── __init__.py ├── colab_codelab.ipynb ├── compile_pb.sh ├── docs ├── executable_specs.md ├── executors.md ├── metadata_storage.md ├── oss_xm_scope.md ├── parameter_controller.md ├── tensorboard.md └── xm_launch_api_principles.md ├── examples ├── cifar10_tensorflow │ ├── cifar10.py │ ├── launcher.py │ └── requirements.txt ├── cifar10_tensorflow_k8s_multiworker │ ├── cifar10.py │ ├── launcher.py │ └── requirements.txt ├── cifar10_tensorflow_k8s_ps │ ├── cifar10.py │ ├── launcher.py │ └── requirements.txt ├── cifar10_tensorflow_k8s_tensorboard │ ├── cifar10.py │ ├── launcher.py │ └── requirements.txt ├── cifar10_tensorflow_tpu │ ├── cifar10.py │ ├── launcher.py │ └── requirements.txt ├── cifar10_torch │ ├── cifar10.py │ ├── launcher.py │ └── requirements.txt ├── cifar10_torch_xla │ ├── cifar10.py │ ├── launcher.py │ └── requirements.txt ├── dockerfile │ ├── Dockerfile │ ├── arg_printer.cc │ └── launcher.py ├── dopamine │ ├── launcher.py │ └── train.py ├── local_arg_printer │ ├── BUILD │ ├── WORKSPACE │ ├── arg_printer.cc │ └── launcher.py ├── local_container_gpu │ ├── cifar10.py │ ├── launcher.py │ └── requirements.txt ├── local_container_links │ ├── BUILD.bazel │ ├── README.md │ ├── WORKSPACE │ ├── launcher.py │ ├── requirements.txt │ └── server.py ├── parameter_controller │ ├── db_config.yaml │ ├── inner_job │ │ ├── requirements.txt │ │ └── wait_job.py │ ├── launcher.py │ └── requirements.txt └── vizier │ ├── README.md │ ├── launcher.py │ ├── polynomial.py │ └── requirements.txt ├── jupyter_codelab.ipynb ├── setup.py ├── setup_scripts ├── install_bazel.sh ├── install_docker.sh ├── install_gcloud.sh ├── install_python.sh ├── install_xmanager.sh ├── setup_all.sh └── setup_gcp.sh └── xmanager ├── bazel ├── client.py └── file_utils.py ├── cli ├── README.md └── cli.py ├── cloud ├── README.md ├── __init__.py ├── auth.py ├── auth_test.py ├── build_image.py ├── build_image_test.py ├── cloud_build.py ├── cloud_build_test.py ├── data │ └── wrapped_entrypoint.sh ├── docker_lib.py ├── kubernetes.py ├── kubernetes_test.py ├── utils.py ├── utils_test.py ├── vertex.py └── vertex_test.py ├── contrib ├── __init__.py ├── addressing.py ├── addressing_test.py ├── copybara.py ├── flow.py ├── framework_defaults.py ├── framework_defaults_test.py ├── gcs.py ├── gcs_test.py ├── parameter_controller.py ├── process_entry.py ├── tensorboard.py ├── tensorboard_test.py ├── tpu.py ├── xm_tensorflow.py └── xm_tensorflow_test.py ├── docker └── docker_adapter.py ├── generated ├── README.md ├── build_event_stream_pb2.py ├── command_line_pb2.py ├── data_pb2.py ├── failure_details_pb2.py ├── invocation_policy_pb2.py └── option_filters_pb2.py ├── module_lazy_loader ├── lazy_loader_module_attrs_test.py └── module_lazy_loader.py ├── vizier └── vizier_cloud │ ├── __init__.py │ ├── study_factory.py │ ├── vizier_controller.py │ ├── vizier_exploration.py │ └── vizier_worker.py ├── xm ├── README.md ├── __init__.py ├── async_packager.py ├── async_packager_test.py ├── compute_units.py ├── core.py ├── core_test.py ├── executables.py ├── executables_test.py ├── id_predictor.py ├── id_predictor_test.py ├── job_blocks.py ├── job_blocks_test.py ├── job_operators.py ├── job_operators_test.py ├── metadata_context.py ├── packagables.py ├── packagables_generator.py ├── packagables_test.py ├── resources.py ├── resources_test.py ├── utils.py └── utils_test.py ├── xm_flags.py ├── xm_local ├── README.md ├── __init__.py ├── executables.py ├── execution.py ├── execution_test.py ├── executors.py ├── executors_test.py ├── experiment.py ├── handles.py ├── multiplexer.py ├── packaging │ ├── bazel_tools.py │ ├── bazel_tools_test.py │ ├── cloud.py │ ├── local.py │ └── router.py ├── registry.py ├── status.py └── storage │ ├── alembic.ini │ ├── alembic │ ├── README.md │ ├── env.py │ ├── script.py.mako │ └── versions │ │ └── f45829405692_migrate_or_create.py │ ├── data.proto │ └── database.py └── xm_mock └── __init__.py /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/__init__.py -------------------------------------------------------------------------------- /colab_codelab.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/colab_codelab.ipynb -------------------------------------------------------------------------------- /compile_pb.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/compile_pb.sh -------------------------------------------------------------------------------- /docs/executable_specs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/docs/executable_specs.md -------------------------------------------------------------------------------- /docs/executors.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/docs/executors.md -------------------------------------------------------------------------------- /docs/metadata_storage.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/docs/metadata_storage.md -------------------------------------------------------------------------------- /docs/oss_xm_scope.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/docs/oss_xm_scope.md -------------------------------------------------------------------------------- /docs/parameter_controller.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/docs/parameter_controller.md -------------------------------------------------------------------------------- /docs/tensorboard.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/docs/tensorboard.md -------------------------------------------------------------------------------- /docs/xm_launch_api_principles.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/docs/xm_launch_api_principles.md -------------------------------------------------------------------------------- /examples/cifar10_tensorflow/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_tensorflow/cifar10.py -------------------------------------------------------------------------------- /examples/cifar10_tensorflow/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_tensorflow/launcher.py -------------------------------------------------------------------------------- /examples/cifar10_tensorflow/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_tensorflow/requirements.txt -------------------------------------------------------------------------------- /examples/cifar10_tensorflow_k8s_multiworker/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_tensorflow_k8s_multiworker/cifar10.py -------------------------------------------------------------------------------- /examples/cifar10_tensorflow_k8s_multiworker/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_tensorflow_k8s_multiworker/launcher.py -------------------------------------------------------------------------------- /examples/cifar10_tensorflow_k8s_multiworker/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_tensorflow_k8s_multiworker/requirements.txt -------------------------------------------------------------------------------- /examples/cifar10_tensorflow_k8s_ps/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_tensorflow_k8s_ps/cifar10.py -------------------------------------------------------------------------------- /examples/cifar10_tensorflow_k8s_ps/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_tensorflow_k8s_ps/launcher.py -------------------------------------------------------------------------------- /examples/cifar10_tensorflow_k8s_ps/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_tensorflow_k8s_ps/requirements.txt -------------------------------------------------------------------------------- /examples/cifar10_tensorflow_k8s_tensorboard/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_tensorflow_k8s_tensorboard/cifar10.py -------------------------------------------------------------------------------- /examples/cifar10_tensorflow_k8s_tensorboard/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_tensorflow_k8s_tensorboard/launcher.py -------------------------------------------------------------------------------- /examples/cifar10_tensorflow_k8s_tensorboard/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_tensorflow_k8s_tensorboard/requirements.txt -------------------------------------------------------------------------------- /examples/cifar10_tensorflow_tpu/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_tensorflow_tpu/cifar10.py -------------------------------------------------------------------------------- /examples/cifar10_tensorflow_tpu/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_tensorflow_tpu/launcher.py -------------------------------------------------------------------------------- /examples/cifar10_tensorflow_tpu/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_tensorflow_tpu/requirements.txt -------------------------------------------------------------------------------- /examples/cifar10_torch/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_torch/cifar10.py -------------------------------------------------------------------------------- /examples/cifar10_torch/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_torch/launcher.py -------------------------------------------------------------------------------- /examples/cifar10_torch/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_torch/requirements.txt -------------------------------------------------------------------------------- /examples/cifar10_torch_xla/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_torch_xla/cifar10.py -------------------------------------------------------------------------------- /examples/cifar10_torch_xla/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_torch_xla/launcher.py -------------------------------------------------------------------------------- /examples/cifar10_torch_xla/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/cifar10_torch_xla/requirements.txt -------------------------------------------------------------------------------- /examples/dockerfile/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/dockerfile/Dockerfile -------------------------------------------------------------------------------- /examples/dockerfile/arg_printer.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/dockerfile/arg_printer.cc -------------------------------------------------------------------------------- /examples/dockerfile/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/dockerfile/launcher.py -------------------------------------------------------------------------------- /examples/dopamine/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/dopamine/launcher.py -------------------------------------------------------------------------------- /examples/dopamine/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/dopamine/train.py -------------------------------------------------------------------------------- /examples/local_arg_printer/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/local_arg_printer/BUILD -------------------------------------------------------------------------------- /examples/local_arg_printer/WORKSPACE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/local_arg_printer/WORKSPACE -------------------------------------------------------------------------------- /examples/local_arg_printer/arg_printer.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/local_arg_printer/arg_printer.cc -------------------------------------------------------------------------------- /examples/local_arg_printer/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/local_arg_printer/launcher.py -------------------------------------------------------------------------------- /examples/local_container_gpu/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/local_container_gpu/cifar10.py -------------------------------------------------------------------------------- /examples/local_container_gpu/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/local_container_gpu/launcher.py -------------------------------------------------------------------------------- /examples/local_container_gpu/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/local_container_gpu/requirements.txt -------------------------------------------------------------------------------- /examples/local_container_links/BUILD.bazel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/local_container_links/BUILD.bazel -------------------------------------------------------------------------------- /examples/local_container_links/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/local_container_links/README.md -------------------------------------------------------------------------------- /examples/local_container_links/WORKSPACE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/local_container_links/WORKSPACE -------------------------------------------------------------------------------- /examples/local_container_links/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/local_container_links/launcher.py -------------------------------------------------------------------------------- /examples/local_container_links/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/local_container_links/requirements.txt -------------------------------------------------------------------------------- /examples/local_container_links/server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/local_container_links/server.py -------------------------------------------------------------------------------- /examples/parameter_controller/db_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/parameter_controller/db_config.yaml -------------------------------------------------------------------------------- /examples/parameter_controller/inner_job/requirements.txt: -------------------------------------------------------------------------------- 1 | absl-py -------------------------------------------------------------------------------- /examples/parameter_controller/inner_job/wait_job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/parameter_controller/inner_job/wait_job.py -------------------------------------------------------------------------------- /examples/parameter_controller/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/parameter_controller/launcher.py -------------------------------------------------------------------------------- /examples/parameter_controller/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/parameter_controller/requirements.txt -------------------------------------------------------------------------------- /examples/vizier/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/vizier/README.md -------------------------------------------------------------------------------- /examples/vizier/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/vizier/launcher.py -------------------------------------------------------------------------------- /examples/vizier/polynomial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/examples/vizier/polynomial.py -------------------------------------------------------------------------------- /examples/vizier/requirements.txt: -------------------------------------------------------------------------------- 1 | absl-py 2 | google-cloud-aiplatform 3 | -------------------------------------------------------------------------------- /jupyter_codelab.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/jupyter_codelab.ipynb -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/setup.py -------------------------------------------------------------------------------- /setup_scripts/install_bazel.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/setup_scripts/install_bazel.sh -------------------------------------------------------------------------------- /setup_scripts/install_docker.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/setup_scripts/install_docker.sh -------------------------------------------------------------------------------- /setup_scripts/install_gcloud.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/setup_scripts/install_gcloud.sh -------------------------------------------------------------------------------- /setup_scripts/install_python.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/setup_scripts/install_python.sh -------------------------------------------------------------------------------- /setup_scripts/install_xmanager.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/setup_scripts/install_xmanager.sh -------------------------------------------------------------------------------- /setup_scripts/setup_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/setup_scripts/setup_all.sh -------------------------------------------------------------------------------- /setup_scripts/setup_gcp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/setup_scripts/setup_gcp.sh -------------------------------------------------------------------------------- /xmanager/bazel/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/bazel/client.py -------------------------------------------------------------------------------- /xmanager/bazel/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/bazel/file_utils.py -------------------------------------------------------------------------------- /xmanager/cli/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cli/README.md -------------------------------------------------------------------------------- /xmanager/cli/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cli/cli.py -------------------------------------------------------------------------------- /xmanager/cloud/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cloud/README.md -------------------------------------------------------------------------------- /xmanager/cloud/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cloud/__init__.py -------------------------------------------------------------------------------- /xmanager/cloud/auth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cloud/auth.py -------------------------------------------------------------------------------- /xmanager/cloud/auth_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cloud/auth_test.py -------------------------------------------------------------------------------- /xmanager/cloud/build_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cloud/build_image.py -------------------------------------------------------------------------------- /xmanager/cloud/build_image_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cloud/build_image_test.py -------------------------------------------------------------------------------- /xmanager/cloud/cloud_build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cloud/cloud_build.py -------------------------------------------------------------------------------- /xmanager/cloud/cloud_build_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cloud/cloud_build_test.py -------------------------------------------------------------------------------- /xmanager/cloud/data/wrapped_entrypoint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cloud/data/wrapped_entrypoint.sh -------------------------------------------------------------------------------- /xmanager/cloud/docker_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cloud/docker_lib.py -------------------------------------------------------------------------------- /xmanager/cloud/kubernetes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cloud/kubernetes.py -------------------------------------------------------------------------------- /xmanager/cloud/kubernetes_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cloud/kubernetes_test.py -------------------------------------------------------------------------------- /xmanager/cloud/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cloud/utils.py -------------------------------------------------------------------------------- /xmanager/cloud/utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cloud/utils_test.py -------------------------------------------------------------------------------- /xmanager/cloud/vertex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cloud/vertex.py -------------------------------------------------------------------------------- /xmanager/cloud/vertex_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/cloud/vertex_test.py -------------------------------------------------------------------------------- /xmanager/contrib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/contrib/__init__.py -------------------------------------------------------------------------------- /xmanager/contrib/addressing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/contrib/addressing.py -------------------------------------------------------------------------------- /xmanager/contrib/addressing_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/contrib/addressing_test.py -------------------------------------------------------------------------------- /xmanager/contrib/copybara.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/contrib/copybara.py -------------------------------------------------------------------------------- /xmanager/contrib/flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/contrib/flow.py -------------------------------------------------------------------------------- /xmanager/contrib/framework_defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/contrib/framework_defaults.py -------------------------------------------------------------------------------- /xmanager/contrib/framework_defaults_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/contrib/framework_defaults_test.py -------------------------------------------------------------------------------- /xmanager/contrib/gcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/contrib/gcs.py -------------------------------------------------------------------------------- /xmanager/contrib/gcs_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/contrib/gcs_test.py -------------------------------------------------------------------------------- /xmanager/contrib/parameter_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/contrib/parameter_controller.py -------------------------------------------------------------------------------- /xmanager/contrib/process_entry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/contrib/process_entry.py -------------------------------------------------------------------------------- /xmanager/contrib/tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/contrib/tensorboard.py -------------------------------------------------------------------------------- /xmanager/contrib/tensorboard_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/contrib/tensorboard_test.py -------------------------------------------------------------------------------- /xmanager/contrib/tpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/contrib/tpu.py -------------------------------------------------------------------------------- /xmanager/contrib/xm_tensorflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/contrib/xm_tensorflow.py -------------------------------------------------------------------------------- /xmanager/contrib/xm_tensorflow_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/contrib/xm_tensorflow_test.py -------------------------------------------------------------------------------- /xmanager/docker/docker_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/docker/docker_adapter.py -------------------------------------------------------------------------------- /xmanager/generated/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/generated/README.md -------------------------------------------------------------------------------- /xmanager/generated/build_event_stream_pb2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/generated/build_event_stream_pb2.py -------------------------------------------------------------------------------- /xmanager/generated/command_line_pb2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/generated/command_line_pb2.py -------------------------------------------------------------------------------- /xmanager/generated/data_pb2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/generated/data_pb2.py -------------------------------------------------------------------------------- /xmanager/generated/failure_details_pb2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/generated/failure_details_pb2.py -------------------------------------------------------------------------------- /xmanager/generated/invocation_policy_pb2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/generated/invocation_policy_pb2.py -------------------------------------------------------------------------------- /xmanager/generated/option_filters_pb2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/generated/option_filters_pb2.py -------------------------------------------------------------------------------- /xmanager/module_lazy_loader/lazy_loader_module_attrs_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/module_lazy_loader/lazy_loader_module_attrs_test.py -------------------------------------------------------------------------------- /xmanager/module_lazy_loader/module_lazy_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/module_lazy_loader/module_lazy_loader.py -------------------------------------------------------------------------------- /xmanager/vizier/vizier_cloud/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/vizier/vizier_cloud/__init__.py -------------------------------------------------------------------------------- /xmanager/vizier/vizier_cloud/study_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/vizier/vizier_cloud/study_factory.py -------------------------------------------------------------------------------- /xmanager/vizier/vizier_cloud/vizier_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/vizier/vizier_cloud/vizier_controller.py -------------------------------------------------------------------------------- /xmanager/vizier/vizier_cloud/vizier_exploration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/vizier/vizier_cloud/vizier_exploration.py -------------------------------------------------------------------------------- /xmanager/vizier/vizier_cloud/vizier_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/vizier/vizier_cloud/vizier_worker.py -------------------------------------------------------------------------------- /xmanager/xm/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/README.md -------------------------------------------------------------------------------- /xmanager/xm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/__init__.py -------------------------------------------------------------------------------- /xmanager/xm/async_packager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/async_packager.py -------------------------------------------------------------------------------- /xmanager/xm/async_packager_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/async_packager_test.py -------------------------------------------------------------------------------- /xmanager/xm/compute_units.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/compute_units.py -------------------------------------------------------------------------------- /xmanager/xm/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/core.py -------------------------------------------------------------------------------- /xmanager/xm/core_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/core_test.py -------------------------------------------------------------------------------- /xmanager/xm/executables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/executables.py -------------------------------------------------------------------------------- /xmanager/xm/executables_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/executables_test.py -------------------------------------------------------------------------------- /xmanager/xm/id_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/id_predictor.py -------------------------------------------------------------------------------- /xmanager/xm/id_predictor_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/id_predictor_test.py -------------------------------------------------------------------------------- /xmanager/xm/job_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/job_blocks.py -------------------------------------------------------------------------------- /xmanager/xm/job_blocks_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/job_blocks_test.py -------------------------------------------------------------------------------- /xmanager/xm/job_operators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/job_operators.py -------------------------------------------------------------------------------- /xmanager/xm/job_operators_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/job_operators_test.py -------------------------------------------------------------------------------- /xmanager/xm/metadata_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/metadata_context.py -------------------------------------------------------------------------------- /xmanager/xm/packagables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/packagables.py -------------------------------------------------------------------------------- /xmanager/xm/packagables_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/packagables_generator.py -------------------------------------------------------------------------------- /xmanager/xm/packagables_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/packagables_test.py -------------------------------------------------------------------------------- /xmanager/xm/resources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/resources.py -------------------------------------------------------------------------------- /xmanager/xm/resources_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/resources_test.py -------------------------------------------------------------------------------- /xmanager/xm/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/utils.py -------------------------------------------------------------------------------- /xmanager/xm/utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm/utils_test.py -------------------------------------------------------------------------------- /xmanager/xm_flags.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_flags.py -------------------------------------------------------------------------------- /xmanager/xm_local/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/README.md -------------------------------------------------------------------------------- /xmanager/xm_local/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/__init__.py -------------------------------------------------------------------------------- /xmanager/xm_local/executables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/executables.py -------------------------------------------------------------------------------- /xmanager/xm_local/execution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/execution.py -------------------------------------------------------------------------------- /xmanager/xm_local/execution_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/execution_test.py -------------------------------------------------------------------------------- /xmanager/xm_local/executors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/executors.py -------------------------------------------------------------------------------- /xmanager/xm_local/executors_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/executors_test.py -------------------------------------------------------------------------------- /xmanager/xm_local/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/experiment.py -------------------------------------------------------------------------------- /xmanager/xm_local/handles.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/handles.py -------------------------------------------------------------------------------- /xmanager/xm_local/multiplexer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/multiplexer.py -------------------------------------------------------------------------------- /xmanager/xm_local/packaging/bazel_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/packaging/bazel_tools.py -------------------------------------------------------------------------------- /xmanager/xm_local/packaging/bazel_tools_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/packaging/bazel_tools_test.py -------------------------------------------------------------------------------- /xmanager/xm_local/packaging/cloud.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/packaging/cloud.py -------------------------------------------------------------------------------- /xmanager/xm_local/packaging/local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/packaging/local.py -------------------------------------------------------------------------------- /xmanager/xm_local/packaging/router.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/packaging/router.py -------------------------------------------------------------------------------- /xmanager/xm_local/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/registry.py -------------------------------------------------------------------------------- /xmanager/xm_local/status.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/status.py -------------------------------------------------------------------------------- /xmanager/xm_local/storage/alembic.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/storage/alembic.ini -------------------------------------------------------------------------------- /xmanager/xm_local/storage/alembic/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/storage/alembic/README.md -------------------------------------------------------------------------------- /xmanager/xm_local/storage/alembic/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/storage/alembic/env.py -------------------------------------------------------------------------------- /xmanager/xm_local/storage/alembic/script.py.mako: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/storage/alembic/script.py.mako -------------------------------------------------------------------------------- /xmanager/xm_local/storage/alembic/versions/f45829405692_migrate_or_create.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/storage/alembic/versions/f45829405692_migrate_or_create.py -------------------------------------------------------------------------------- /xmanager/xm_local/storage/data.proto: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/storage/data.proto -------------------------------------------------------------------------------- /xmanager/xm_local/storage/database.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_local/storage/database.py -------------------------------------------------------------------------------- /xmanager/xm_mock/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/xmanager/HEAD/xmanager/xm_mock/__init__.py --------------------------------------------------------------------------------