├── .coveragerc ├── .dockerignore ├── .flake8 ├── .github ├── ISSUE_TEMPLATE │ ├── bug-report.md │ ├── documentation.md │ ├── feature-request.md │ └── questions-help-support.md ├── PULL_REQUEST_TEMPLATE │ └── pull_request_template.md ├── pull_request_template.md └── workflows │ ├── aws-batch-integration-tests.yaml │ ├── components-integration-tests.yaml │ ├── container.yaml │ ├── doc-build.yaml │ ├── kubernetes-minikube-integration-test.yaml │ ├── lint.yaml │ ├── nightly.yaml │ ├── pyre.yaml │ ├── python-unittests.yaml │ └── slurm-local-integration-tests.yaml ├── .gitignore ├── .lintrunner.toml ├── .pyre_configuration ├── .torchxconfig ├── .torchxignore ├── .watchmanconfig ├── CHANGELOG.md ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── dev-requirements.txt ├── docs ├── .gitignore ├── Makefile ├── doc_push.sh ├── papermill.sh ├── requirements.txt ├── source │ ├── .gitignore │ ├── _static │ │ ├── css │ │ │ └── torchx.css │ │ ├── img │ │ │ ├── efs-setup.jpg │ │ │ ├── gallery-app.png │ │ │ ├── gallery-component.png │ │ │ ├── gallery-kfp.png │ │ │ ├── gallery-lib.png │ │ │ ├── pytorch-logo-dark.svg │ │ │ ├── pytorch-logo-flame.png │ │ │ └── torchx.png │ │ └── js │ │ │ └── torchx.js │ ├── _templates │ │ └── layout.html │ ├── advanced.rst │ ├── app_best_practices.rst │ ├── basics.rst │ ├── cli.rst │ ├── component_best_practices.rst │ ├── components │ │ ├── components_diagram.jpg │ │ ├── distributed.rst │ │ ├── hpo_diagram.png │ │ ├── interpret.rst │ │ ├── metrics.rst │ │ ├── overview.rst │ │ ├── serve.rst │ │ ├── train.rst │ │ └── utils.rst │ ├── conf.py │ ├── custom_components.md │ ├── examples │ │ └── .gitignore │ ├── ext │ │ ├── compatibility.py │ │ ├── fbcode.py │ │ └── runopts.py │ ├── index.rst │ ├── quickstart.md │ ├── runner.config.rst │ ├── runner.rst │ ├── runner_diagram.png │ ├── runtime │ │ ├── overview.rst │ │ └── tracking.rst │ ├── scheduler_diagram.png │ ├── schedulers.rst │ ├── schedulers │ │ ├── aws_batch.rst │ │ ├── aws_sagemaker.rst │ │ ├── docker.rst │ │ ├── kubernetes.rst │ │ ├── kubernetes_mcad.rst │ │ ├── local.rst │ │ ├── lsf.rst │ │ └── slurm.rst │ ├── specs.rst │ ├── torchx_dev_phases.jpg │ ├── torchx_index_diag.png │ ├── torchx_module_uml.jpg │ ├── tracker.rst │ └── workspace.rst └── versions_html.py ├── fmt-requirements.txt ├── pyproject.toml ├── requirements.txt ├── resources ├── README.md ├── etcd.yaml ├── kfp_volcano_role_binding.yaml ├── torchx-dev-eks-template.yml └── torchx_integration_tests_role.yaml ├── scripts ├── awsbatchint.sh ├── collect_env.py ├── component_integration_tests.py ├── example_app_defs.py ├── integ_test_utils.py ├── lint.sh ├── minikube_trainer.py ├── push_nightly.sh ├── pyre.sh ├── setup_minikube.sh ├── setup_pyre.sh ├── slurmdockerint.sh ├── slurmint.sh ├── slurmtest.sh ├── spellcheck.sh └── spellcheck_conf │ ├── spellcheck.yaml │ └── wordlist.txt ├── setup.py ├── stubs └── filelock.pyi ├── tools └── linter │ └── adapters │ ├── pyre_linter.py │ └── ufmt_linter.py ├── torchx ├── __init__.py ├── _version.py ├── apps │ ├── __init__.py │ ├── serve │ │ ├── __init__.py │ │ ├── serve.py │ │ └── test │ │ │ ├── __init__.py │ │ │ └── serve_test.py │ └── utils │ │ ├── __init__.py │ │ ├── booth_main.py │ │ ├── copy_main.py │ │ ├── process_monitor.py │ │ └── test │ │ ├── __init__.py │ │ ├── booth_test.py │ │ ├── copy_test.py │ │ └── process_monitor_test.py ├── cli │ ├── __init__.py │ ├── argparse_util.py │ ├── cmd_base.py │ ├── cmd_cancel.py │ ├── cmd_configure.py │ ├── cmd_delete.py │ ├── cmd_describe.py │ ├── cmd_list.py │ ├── cmd_log.py │ ├── cmd_run.py │ ├── cmd_runopts.py │ ├── cmd_status.py │ ├── cmd_tracker.py │ ├── colors.py │ ├── main.py │ └── test │ │ ├── __init__.py │ │ ├── argparse_util_test.py │ │ ├── cmd_cancel_test.py │ │ ├── cmd_configure_test.py │ │ ├── cmd_delete_test.py │ │ ├── cmd_describe_test.py │ │ ├── cmd_list_test.py │ │ ├── cmd_log_test.py │ │ ├── cmd_run_test.py │ │ ├── cmd_runopts_test.py │ │ ├── cmd_status_test.py │ │ ├── cmd_tracker_test.py │ │ ├── components.py │ │ ├── container │ │ ├── ps_main.py │ │ ├── reader_main.py │ │ └── train_main.py │ │ └── main_test.py ├── components │ ├── README.md │ ├── __init__.py │ ├── component_test_base.py │ ├── dist.py │ ├── integration_tests │ │ ├── __init__.py │ │ ├── component_provider.py │ │ ├── integ_tests.py │ │ └── test │ │ │ ├── __init__.py │ │ │ └── dummy_app.py │ ├── interpret.py │ ├── metrics.py │ ├── serve.py │ ├── structured_arg.py │ ├── test │ │ ├── __init__.py │ │ ├── dist_test.py │ │ ├── metrics_test.py │ │ ├── serve_test.py │ │ ├── structured_arg_test.py │ │ └── utils_test.py │ ├── train.py │ └── utils.py ├── distributed │ ├── __init__.py │ └── test │ │ ├── __init__.py │ │ └── dist_test.py ├── examples │ ├── README.rst │ ├── __init__.py │ ├── apps │ │ ├── .torchxconfig │ │ ├── README.rst │ │ ├── __init__.py │ │ ├── compute_world_size │ │ │ ├── README.rst │ │ │ ├── config │ │ │ │ ├── __init__.py │ │ │ │ └── defaults.yaml │ │ │ ├── main.py │ │ │ └── module │ │ │ │ ├── test │ │ │ │ └── util_test.py │ │ │ │ └── util.py │ │ ├── datapreproc │ │ │ ├── README.rst │ │ │ ├── __init__.py │ │ │ └── datapreproc.py │ │ ├── lightning │ │ │ ├── README.rst │ │ │ ├── __init__.py │ │ │ ├── data.py │ │ │ ├── handler │ │ │ │ └── handler.py │ │ │ ├── interpret.py │ │ │ ├── model.py │ │ │ ├── profiler.py │ │ │ ├── test │ │ │ │ ├── __init__.py │ │ │ │ └── model_test.py │ │ │ └── train.py │ │ └── tracker │ │ │ ├── .torchxconfig │ │ │ ├── fsspec_backend.conf │ │ │ ├── main.py │ │ │ └── tracker_example.dist-info │ │ │ └── entry_points.txt │ └── torchx_out_of_sync_training.py ├── notebook.py ├── pipelines │ └── __init__.py ├── runner │ ├── __init__.py │ ├── api.py │ ├── config.py │ ├── events │ │ ├── __init__.py │ │ ├── api.py │ │ ├── handlers.py │ │ └── test │ │ │ └── lib_test.py │ └── test │ │ ├── __init__.py │ │ ├── api_test.py │ │ ├── config_test.py │ │ └── resource │ │ └── distributed.py ├── runtime │ ├── __init__.py │ ├── container │ │ ├── Dockerfile │ │ └── build.sh │ └── tracking │ │ ├── __init__.py │ │ ├── api.py │ │ └── test │ │ ├── __init__.py │ │ └── api_test.py ├── schedulers │ ├── __init__.py │ ├── api.py │ ├── aws_batch_scheduler.py │ ├── aws_sagemaker_scheduler.py │ ├── devices.py │ ├── docker_scheduler.py │ ├── ids.py │ ├── kubernetes_mcad_scheduler.py │ ├── kubernetes_scheduler.py │ ├── local_scheduler.py │ ├── lsf_scheduler.py │ ├── slurm_scheduler.py │ ├── streams.py │ └── test │ │ ├── __init__.py │ │ ├── api_test.py │ │ ├── aws_batch_scheduler_test.py │ │ ├── aws_sagemaker_scheduler_test.py │ │ ├── devices_test.py │ │ ├── docker_scheduler_test.py │ │ ├── ids_test.py │ │ ├── kubernetes_mcad_scheduler_test.py │ │ ├── kubernetes_scheduler_test.py │ │ ├── local_scheduler_test.py │ │ ├── lsf_scheduler_test.py │ │ ├── registry_test.py │ │ ├── slurm-squeue-output.json │ │ ├── slurm_scheduler_test.py │ │ ├── streams_test.py │ │ ├── test_util.py │ │ └── train.py ├── specs │ ├── __init__.py │ ├── api.py │ ├── builders.py │ ├── file_linter.py │ ├── finder.py │ ├── named_resources_aws.py │ ├── named_resources_generic.py │ └── test │ │ ├── __init__.py │ │ ├── api_test.py │ │ ├── builders_test.py │ │ ├── components │ │ ├── __init__.py │ │ ├── a │ │ │ ├── __init__.py │ │ │ └── b │ │ │ │ ├── __init__.py │ │ │ │ └── c.py │ │ └── c │ │ │ ├── __init__.py │ │ │ └── d.py │ │ ├── file_linter_test.py │ │ ├── finder_test.py │ │ ├── named_resources_aws_test.py │ │ ├── named_resources_generic_test.py │ │ └── named_resources_test.py ├── test │ ├── __init__.py │ ├── fixtures.py │ ├── minio.py │ ├── notebook_test.py │ ├── suites.py │ └── version_test.py ├── tracker │ ├── __init__.py │ ├── api.py │ ├── backend │ │ ├── __init__.py │ │ ├── fsspec.py │ │ └── test │ │ │ ├── __init__.py │ │ │ └── fsspec_test.py │ ├── mlflow.py │ └── test │ │ ├── __init__.py │ │ ├── api_test.py │ │ └── mlflow_test.py ├── util │ ├── __init__.py │ ├── cuda.py │ ├── datetime.py │ ├── entrypoints.py │ ├── io.py │ ├── log_tee_helpers.py │ ├── modules.py │ ├── session.py │ ├── shlex.py │ ├── strings.py │ ├── test │ │ ├── __init__.py │ │ ├── cuda_test.py │ │ ├── entrypoints_test.py │ │ ├── modules_test.py │ │ ├── shlex_test.py │ │ ├── strings_test.py │ │ └── types_test.py │ └── types.py ├── version.py └── workspace │ ├── __init__.py │ ├── api.py │ ├── dir_workspace.py │ ├── docker_workspace.py │ └── test │ ├── __init__.py │ ├── api_test.py │ ├── dir_workspace_test.py │ ├── docker_workspace_test.py │ ├── expected_print_push_events.txt │ └── mock_docker_push_events.json └── typestubs └── torch └── monitor.pyi /.coveragerc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.coveragerc -------------------------------------------------------------------------------- /.dockerignore: -------------------------------------------------------------------------------- 1 | .git 2 | .idea 3 | .pyre 4 | docs 5 | -------------------------------------------------------------------------------- /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug-report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.github/ISSUE_TEMPLATE/bug-report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/documentation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.github/ISSUE_TEMPLATE/documentation.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature-request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.github/ISSUE_TEMPLATE/feature-request.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/questions-help-support.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.github/ISSUE_TEMPLATE/questions-help-support.md -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE/pull_request_template.md: -------------------------------------------------------------------------------- 1 | Fixes #{issue number} 2 | -------------------------------------------------------------------------------- /.github/pull_request_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.github/pull_request_template.md -------------------------------------------------------------------------------- /.github/workflows/aws-batch-integration-tests.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.github/workflows/aws-batch-integration-tests.yaml -------------------------------------------------------------------------------- /.github/workflows/components-integration-tests.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.github/workflows/components-integration-tests.yaml -------------------------------------------------------------------------------- /.github/workflows/container.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.github/workflows/container.yaml -------------------------------------------------------------------------------- /.github/workflows/doc-build.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.github/workflows/doc-build.yaml -------------------------------------------------------------------------------- /.github/workflows/kubernetes-minikube-integration-test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.github/workflows/kubernetes-minikube-integration-test.yaml -------------------------------------------------------------------------------- /.github/workflows/lint.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.github/workflows/lint.yaml -------------------------------------------------------------------------------- /.github/workflows/nightly.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.github/workflows/nightly.yaml -------------------------------------------------------------------------------- /.github/workflows/pyre.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.github/workflows/pyre.yaml -------------------------------------------------------------------------------- /.github/workflows/python-unittests.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.github/workflows/python-unittests.yaml -------------------------------------------------------------------------------- /.github/workflows/slurm-local-integration-tests.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.github/workflows/slurm-local-integration-tests.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.gitignore -------------------------------------------------------------------------------- /.lintrunner.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.lintrunner.toml -------------------------------------------------------------------------------- /.pyre_configuration: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.pyre_configuration -------------------------------------------------------------------------------- /.torchxconfig: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.torchxconfig -------------------------------------------------------------------------------- /.torchxignore: -------------------------------------------------------------------------------- 1 | .dockerignore -------------------------------------------------------------------------------- /.watchmanconfig: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/.watchmanconfig -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/README.md -------------------------------------------------------------------------------- /dev-requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/dev-requirements.txt -------------------------------------------------------------------------------- /docs/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/.gitignore -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/doc_push.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/doc_push.sh -------------------------------------------------------------------------------- /docs/papermill.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/papermill.sh -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/source/.gitignore: -------------------------------------------------------------------------------- 1 | .torchxconfig 2 | Dockerfile* 3 | *.py 4 | -------------------------------------------------------------------------------- /docs/source/_static/css/torchx.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/_static/css/torchx.css -------------------------------------------------------------------------------- /docs/source/_static/img/efs-setup.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/_static/img/efs-setup.jpg -------------------------------------------------------------------------------- /docs/source/_static/img/gallery-app.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/_static/img/gallery-app.png -------------------------------------------------------------------------------- /docs/source/_static/img/gallery-component.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/_static/img/gallery-component.png -------------------------------------------------------------------------------- /docs/source/_static/img/gallery-kfp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/_static/img/gallery-kfp.png -------------------------------------------------------------------------------- /docs/source/_static/img/gallery-lib.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/_static/img/gallery-lib.png -------------------------------------------------------------------------------- /docs/source/_static/img/pytorch-logo-dark.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/_static/img/pytorch-logo-dark.svg -------------------------------------------------------------------------------- /docs/source/_static/img/pytorch-logo-flame.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/_static/img/pytorch-logo-flame.png -------------------------------------------------------------------------------- /docs/source/_static/img/torchx.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/_static/img/torchx.png -------------------------------------------------------------------------------- /docs/source/_static/js/torchx.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/_static/js/torchx.js -------------------------------------------------------------------------------- /docs/source/_templates/layout.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/_templates/layout.html -------------------------------------------------------------------------------- /docs/source/advanced.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/advanced.rst -------------------------------------------------------------------------------- /docs/source/app_best_practices.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/app_best_practices.rst -------------------------------------------------------------------------------- /docs/source/basics.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/basics.rst -------------------------------------------------------------------------------- /docs/source/cli.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/cli.rst -------------------------------------------------------------------------------- /docs/source/component_best_practices.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/component_best_practices.rst -------------------------------------------------------------------------------- /docs/source/components/components_diagram.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/components/components_diagram.jpg -------------------------------------------------------------------------------- /docs/source/components/distributed.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/components/distributed.rst -------------------------------------------------------------------------------- /docs/source/components/hpo_diagram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/components/hpo_diagram.png -------------------------------------------------------------------------------- /docs/source/components/interpret.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/components/interpret.rst -------------------------------------------------------------------------------- /docs/source/components/metrics.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/components/metrics.rst -------------------------------------------------------------------------------- /docs/source/components/overview.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/components/overview.rst -------------------------------------------------------------------------------- /docs/source/components/serve.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/components/serve.rst -------------------------------------------------------------------------------- /docs/source/components/train.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/components/train.rst -------------------------------------------------------------------------------- /docs/source/components/utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/components/utils.rst -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/custom_components.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/custom_components.md -------------------------------------------------------------------------------- /docs/source/examples/.gitignore: -------------------------------------------------------------------------------- 1 | *.py 2 | Dockerfile 3 | -------------------------------------------------------------------------------- /docs/source/ext/compatibility.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/ext/compatibility.py -------------------------------------------------------------------------------- /docs/source/ext/fbcode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/ext/fbcode.py -------------------------------------------------------------------------------- /docs/source/ext/runopts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/ext/runopts.py -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/quickstart.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/quickstart.md -------------------------------------------------------------------------------- /docs/source/runner.config.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/runner.config.rst -------------------------------------------------------------------------------- /docs/source/runner.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/runner.rst -------------------------------------------------------------------------------- /docs/source/runner_diagram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/runner_diagram.png -------------------------------------------------------------------------------- /docs/source/runtime/overview.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/runtime/overview.rst -------------------------------------------------------------------------------- /docs/source/runtime/tracking.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/runtime/tracking.rst -------------------------------------------------------------------------------- /docs/source/scheduler_diagram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/scheduler_diagram.png -------------------------------------------------------------------------------- /docs/source/schedulers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/schedulers.rst -------------------------------------------------------------------------------- /docs/source/schedulers/aws_batch.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/schedulers/aws_batch.rst -------------------------------------------------------------------------------- /docs/source/schedulers/aws_sagemaker.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/schedulers/aws_sagemaker.rst -------------------------------------------------------------------------------- /docs/source/schedulers/docker.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/schedulers/docker.rst -------------------------------------------------------------------------------- /docs/source/schedulers/kubernetes.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/schedulers/kubernetes.rst -------------------------------------------------------------------------------- /docs/source/schedulers/kubernetes_mcad.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/schedulers/kubernetes_mcad.rst -------------------------------------------------------------------------------- /docs/source/schedulers/local.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/schedulers/local.rst -------------------------------------------------------------------------------- /docs/source/schedulers/lsf.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/schedulers/lsf.rst -------------------------------------------------------------------------------- /docs/source/schedulers/slurm.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/schedulers/slurm.rst -------------------------------------------------------------------------------- /docs/source/specs.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/specs.rst -------------------------------------------------------------------------------- /docs/source/torchx_dev_phases.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/torchx_dev_phases.jpg -------------------------------------------------------------------------------- /docs/source/torchx_index_diag.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/torchx_index_diag.png -------------------------------------------------------------------------------- /docs/source/torchx_module_uml.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/torchx_module_uml.jpg -------------------------------------------------------------------------------- /docs/source/tracker.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/tracker.rst -------------------------------------------------------------------------------- /docs/source/workspace.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/source/workspace.rst -------------------------------------------------------------------------------- /docs/versions_html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/docs/versions_html.py -------------------------------------------------------------------------------- /fmt-requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/fmt-requirements.txt -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/requirements.txt -------------------------------------------------------------------------------- /resources/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/resources/README.md -------------------------------------------------------------------------------- /resources/etcd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/resources/etcd.yaml -------------------------------------------------------------------------------- /resources/kfp_volcano_role_binding.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/resources/kfp_volcano_role_binding.yaml -------------------------------------------------------------------------------- /resources/torchx-dev-eks-template.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/resources/torchx-dev-eks-template.yml -------------------------------------------------------------------------------- /resources/torchx_integration_tests_role.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/resources/torchx_integration_tests_role.yaml -------------------------------------------------------------------------------- /scripts/awsbatchint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/awsbatchint.sh -------------------------------------------------------------------------------- /scripts/collect_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/collect_env.py -------------------------------------------------------------------------------- /scripts/component_integration_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/component_integration_tests.py -------------------------------------------------------------------------------- /scripts/example_app_defs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/example_app_defs.py -------------------------------------------------------------------------------- /scripts/integ_test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/integ_test_utils.py -------------------------------------------------------------------------------- /scripts/lint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/lint.sh -------------------------------------------------------------------------------- /scripts/minikube_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/minikube_trainer.py -------------------------------------------------------------------------------- /scripts/push_nightly.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/push_nightly.sh -------------------------------------------------------------------------------- /scripts/pyre.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/pyre.sh -------------------------------------------------------------------------------- /scripts/setup_minikube.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/setup_minikube.sh -------------------------------------------------------------------------------- /scripts/setup_pyre.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/setup_pyre.sh -------------------------------------------------------------------------------- /scripts/slurmdockerint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/slurmdockerint.sh -------------------------------------------------------------------------------- /scripts/slurmint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/slurmint.sh -------------------------------------------------------------------------------- /scripts/slurmtest.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/slurmtest.sh -------------------------------------------------------------------------------- /scripts/spellcheck.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/spellcheck.sh -------------------------------------------------------------------------------- /scripts/spellcheck_conf/spellcheck.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/spellcheck_conf/spellcheck.yaml -------------------------------------------------------------------------------- /scripts/spellcheck_conf/wordlist.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/scripts/spellcheck_conf/wordlist.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/setup.py -------------------------------------------------------------------------------- /stubs/filelock.pyi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/stubs/filelock.pyi -------------------------------------------------------------------------------- /tools/linter/adapters/pyre_linter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/tools/linter/adapters/pyre_linter.py -------------------------------------------------------------------------------- /tools/linter/adapters/ufmt_linter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/tools/linter/adapters/ufmt_linter.py -------------------------------------------------------------------------------- /torchx/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/__init__.py -------------------------------------------------------------------------------- /torchx/_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/_version.py -------------------------------------------------------------------------------- /torchx/apps/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/apps/__init__.py -------------------------------------------------------------------------------- /torchx/apps/serve/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/apps/serve/__init__.py -------------------------------------------------------------------------------- /torchx/apps/serve/serve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/apps/serve/serve.py -------------------------------------------------------------------------------- /torchx/apps/serve/test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchx/apps/serve/test/serve_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/apps/serve/test/serve_test.py -------------------------------------------------------------------------------- /torchx/apps/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/apps/utils/__init__.py -------------------------------------------------------------------------------- /torchx/apps/utils/booth_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/apps/utils/booth_main.py -------------------------------------------------------------------------------- /torchx/apps/utils/copy_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/apps/utils/copy_main.py -------------------------------------------------------------------------------- /torchx/apps/utils/process_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/apps/utils/process_monitor.py -------------------------------------------------------------------------------- /torchx/apps/utils/test/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/apps/utils/test/__init__.py -------------------------------------------------------------------------------- /torchx/apps/utils/test/booth_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/apps/utils/test/booth_test.py -------------------------------------------------------------------------------- /torchx/apps/utils/test/copy_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/apps/utils/test/copy_test.py -------------------------------------------------------------------------------- /torchx/apps/utils/test/process_monitor_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/apps/utils/test/process_monitor_test.py -------------------------------------------------------------------------------- /torchx/cli/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/__init__.py -------------------------------------------------------------------------------- /torchx/cli/argparse_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/argparse_util.py -------------------------------------------------------------------------------- /torchx/cli/cmd_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/cmd_base.py -------------------------------------------------------------------------------- /torchx/cli/cmd_cancel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/cmd_cancel.py -------------------------------------------------------------------------------- /torchx/cli/cmd_configure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/cmd_configure.py -------------------------------------------------------------------------------- /torchx/cli/cmd_delete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/cmd_delete.py -------------------------------------------------------------------------------- /torchx/cli/cmd_describe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/cmd_describe.py -------------------------------------------------------------------------------- /torchx/cli/cmd_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/cmd_list.py -------------------------------------------------------------------------------- /torchx/cli/cmd_log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/cmd_log.py -------------------------------------------------------------------------------- /torchx/cli/cmd_run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/cmd_run.py -------------------------------------------------------------------------------- /torchx/cli/cmd_runopts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/cmd_runopts.py -------------------------------------------------------------------------------- /torchx/cli/cmd_status.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/cmd_status.py -------------------------------------------------------------------------------- /torchx/cli/cmd_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/cmd_tracker.py -------------------------------------------------------------------------------- /torchx/cli/colors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/colors.py -------------------------------------------------------------------------------- /torchx/cli/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/main.py -------------------------------------------------------------------------------- /torchx/cli/test/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/__init__.py -------------------------------------------------------------------------------- /torchx/cli/test/argparse_util_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/argparse_util_test.py -------------------------------------------------------------------------------- /torchx/cli/test/cmd_cancel_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/cmd_cancel_test.py -------------------------------------------------------------------------------- /torchx/cli/test/cmd_configure_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/cmd_configure_test.py -------------------------------------------------------------------------------- /torchx/cli/test/cmd_delete_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/cmd_delete_test.py -------------------------------------------------------------------------------- /torchx/cli/test/cmd_describe_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/cmd_describe_test.py -------------------------------------------------------------------------------- /torchx/cli/test/cmd_list_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/cmd_list_test.py -------------------------------------------------------------------------------- /torchx/cli/test/cmd_log_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/cmd_log_test.py -------------------------------------------------------------------------------- /torchx/cli/test/cmd_run_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/cmd_run_test.py -------------------------------------------------------------------------------- /torchx/cli/test/cmd_runopts_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/cmd_runopts_test.py -------------------------------------------------------------------------------- /torchx/cli/test/cmd_status_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/cmd_status_test.py -------------------------------------------------------------------------------- /torchx/cli/test/cmd_tracker_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/cmd_tracker_test.py -------------------------------------------------------------------------------- /torchx/cli/test/components.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/components.py -------------------------------------------------------------------------------- /torchx/cli/test/container/ps_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/container/ps_main.py -------------------------------------------------------------------------------- /torchx/cli/test/container/reader_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/container/reader_main.py -------------------------------------------------------------------------------- /torchx/cli/test/container/train_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/container/train_main.py -------------------------------------------------------------------------------- /torchx/cli/test/main_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/cli/test/main_test.py -------------------------------------------------------------------------------- /torchx/components/README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchx/components/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/__init__.py -------------------------------------------------------------------------------- /torchx/components/component_test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/component_test_base.py -------------------------------------------------------------------------------- /torchx/components/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/dist.py -------------------------------------------------------------------------------- /torchx/components/integration_tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/integration_tests/__init__.py -------------------------------------------------------------------------------- /torchx/components/integration_tests/component_provider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/integration_tests/component_provider.py -------------------------------------------------------------------------------- /torchx/components/integration_tests/integ_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/integration_tests/integ_tests.py -------------------------------------------------------------------------------- /torchx/components/integration_tests/test/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/integration_tests/test/__init__.py -------------------------------------------------------------------------------- /torchx/components/integration_tests/test/dummy_app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/integration_tests/test/dummy_app.py -------------------------------------------------------------------------------- /torchx/components/interpret.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/interpret.py -------------------------------------------------------------------------------- /torchx/components/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/metrics.py -------------------------------------------------------------------------------- /torchx/components/serve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/serve.py -------------------------------------------------------------------------------- /torchx/components/structured_arg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/structured_arg.py -------------------------------------------------------------------------------- /torchx/components/test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchx/components/test/dist_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/test/dist_test.py -------------------------------------------------------------------------------- /torchx/components/test/metrics_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/test/metrics_test.py -------------------------------------------------------------------------------- /torchx/components/test/serve_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/test/serve_test.py -------------------------------------------------------------------------------- /torchx/components/test/structured_arg_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/test/structured_arg_test.py -------------------------------------------------------------------------------- /torchx/components/test/utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/test/utils_test.py -------------------------------------------------------------------------------- /torchx/components/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/train.py -------------------------------------------------------------------------------- /torchx/components/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/components/utils.py -------------------------------------------------------------------------------- /torchx/distributed/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/distributed/__init__.py -------------------------------------------------------------------------------- /torchx/distributed/test/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/distributed/test/__init__.py -------------------------------------------------------------------------------- /torchx/distributed/test/dist_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/distributed/test/dist_test.py -------------------------------------------------------------------------------- /torchx/examples/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/README.rst -------------------------------------------------------------------------------- /torchx/examples/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchx/examples/apps/.torchxconfig: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/.torchxconfig -------------------------------------------------------------------------------- /torchx/examples/apps/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/README.rst -------------------------------------------------------------------------------- /torchx/examples/apps/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchx/examples/apps/compute_world_size/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/compute_world_size/README.rst -------------------------------------------------------------------------------- /torchx/examples/apps/compute_world_size/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/compute_world_size/config/__init__.py -------------------------------------------------------------------------------- /torchx/examples/apps/compute_world_size/config/defaults.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/compute_world_size/config/defaults.yaml -------------------------------------------------------------------------------- /torchx/examples/apps/compute_world_size/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/compute_world_size/main.py -------------------------------------------------------------------------------- /torchx/examples/apps/compute_world_size/module/test/util_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/compute_world_size/module/test/util_test.py -------------------------------------------------------------------------------- /torchx/examples/apps/compute_world_size/module/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/compute_world_size/module/util.py -------------------------------------------------------------------------------- /torchx/examples/apps/datapreproc/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/datapreproc/README.rst -------------------------------------------------------------------------------- /torchx/examples/apps/datapreproc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchx/examples/apps/datapreproc/datapreproc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/datapreproc/datapreproc.py -------------------------------------------------------------------------------- /torchx/examples/apps/lightning/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/lightning/README.rst -------------------------------------------------------------------------------- /torchx/examples/apps/lightning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchx/examples/apps/lightning/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/lightning/data.py -------------------------------------------------------------------------------- /torchx/examples/apps/lightning/handler/handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/lightning/handler/handler.py -------------------------------------------------------------------------------- /torchx/examples/apps/lightning/interpret.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/lightning/interpret.py -------------------------------------------------------------------------------- /torchx/examples/apps/lightning/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/lightning/model.py -------------------------------------------------------------------------------- /torchx/examples/apps/lightning/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/lightning/profiler.py -------------------------------------------------------------------------------- /torchx/examples/apps/lightning/test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchx/examples/apps/lightning/test/model_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/lightning/test/model_test.py -------------------------------------------------------------------------------- /torchx/examples/apps/lightning/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/lightning/train.py -------------------------------------------------------------------------------- /torchx/examples/apps/tracker/.torchxconfig: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/tracker/.torchxconfig -------------------------------------------------------------------------------- /torchx/examples/apps/tracker/fsspec_backend.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/tracker/fsspec_backend.conf -------------------------------------------------------------------------------- /torchx/examples/apps/tracker/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/tracker/main.py -------------------------------------------------------------------------------- /torchx/examples/apps/tracker/tracker_example.dist-info/entry_points.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/apps/tracker/tracker_example.dist-info/entry_points.txt -------------------------------------------------------------------------------- /torchx/examples/torchx_out_of_sync_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/examples/torchx_out_of_sync_training.py -------------------------------------------------------------------------------- /torchx/notebook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/notebook.py -------------------------------------------------------------------------------- /torchx/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/pipelines/__init__.py -------------------------------------------------------------------------------- /torchx/runner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runner/__init__.py -------------------------------------------------------------------------------- /torchx/runner/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runner/api.py -------------------------------------------------------------------------------- /torchx/runner/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runner/config.py -------------------------------------------------------------------------------- /torchx/runner/events/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runner/events/__init__.py -------------------------------------------------------------------------------- /torchx/runner/events/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runner/events/api.py -------------------------------------------------------------------------------- /torchx/runner/events/handlers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runner/events/handlers.py -------------------------------------------------------------------------------- /torchx/runner/events/test/lib_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runner/events/test/lib_test.py -------------------------------------------------------------------------------- /torchx/runner/test/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runner/test/__init__.py -------------------------------------------------------------------------------- /torchx/runner/test/api_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runner/test/api_test.py -------------------------------------------------------------------------------- /torchx/runner/test/config_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runner/test/config_test.py -------------------------------------------------------------------------------- /torchx/runner/test/resource/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runner/test/resource/distributed.py -------------------------------------------------------------------------------- /torchx/runtime/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runtime/__init__.py -------------------------------------------------------------------------------- /torchx/runtime/container/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runtime/container/Dockerfile -------------------------------------------------------------------------------- /torchx/runtime/container/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runtime/container/build.sh -------------------------------------------------------------------------------- /torchx/runtime/tracking/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runtime/tracking/__init__.py -------------------------------------------------------------------------------- /torchx/runtime/tracking/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runtime/tracking/api.py -------------------------------------------------------------------------------- /torchx/runtime/tracking/test/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runtime/tracking/test/__init__.py -------------------------------------------------------------------------------- /torchx/runtime/tracking/test/api_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/runtime/tracking/test/api_test.py -------------------------------------------------------------------------------- /torchx/schedulers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/__init__.py -------------------------------------------------------------------------------- /torchx/schedulers/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/api.py -------------------------------------------------------------------------------- /torchx/schedulers/aws_batch_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/aws_batch_scheduler.py -------------------------------------------------------------------------------- /torchx/schedulers/aws_sagemaker_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/aws_sagemaker_scheduler.py -------------------------------------------------------------------------------- /torchx/schedulers/devices.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/devices.py -------------------------------------------------------------------------------- /torchx/schedulers/docker_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/docker_scheduler.py -------------------------------------------------------------------------------- /torchx/schedulers/ids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/ids.py -------------------------------------------------------------------------------- /torchx/schedulers/kubernetes_mcad_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/kubernetes_mcad_scheduler.py -------------------------------------------------------------------------------- /torchx/schedulers/kubernetes_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/kubernetes_scheduler.py -------------------------------------------------------------------------------- /torchx/schedulers/local_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/local_scheduler.py -------------------------------------------------------------------------------- /torchx/schedulers/lsf_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/lsf_scheduler.py -------------------------------------------------------------------------------- /torchx/schedulers/slurm_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/slurm_scheduler.py -------------------------------------------------------------------------------- /torchx/schedulers/streams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/streams.py -------------------------------------------------------------------------------- /torchx/schedulers/test/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/__init__.py -------------------------------------------------------------------------------- /torchx/schedulers/test/api_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/api_test.py -------------------------------------------------------------------------------- /torchx/schedulers/test/aws_batch_scheduler_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/aws_batch_scheduler_test.py -------------------------------------------------------------------------------- /torchx/schedulers/test/aws_sagemaker_scheduler_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/aws_sagemaker_scheduler_test.py -------------------------------------------------------------------------------- /torchx/schedulers/test/devices_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/devices_test.py -------------------------------------------------------------------------------- /torchx/schedulers/test/docker_scheduler_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/docker_scheduler_test.py -------------------------------------------------------------------------------- /torchx/schedulers/test/ids_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/ids_test.py -------------------------------------------------------------------------------- /torchx/schedulers/test/kubernetes_mcad_scheduler_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/kubernetes_mcad_scheduler_test.py -------------------------------------------------------------------------------- /torchx/schedulers/test/kubernetes_scheduler_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/kubernetes_scheduler_test.py -------------------------------------------------------------------------------- /torchx/schedulers/test/local_scheduler_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/local_scheduler_test.py -------------------------------------------------------------------------------- /torchx/schedulers/test/lsf_scheduler_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/lsf_scheduler_test.py -------------------------------------------------------------------------------- /torchx/schedulers/test/registry_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/registry_test.py -------------------------------------------------------------------------------- /torchx/schedulers/test/slurm-squeue-output.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/slurm-squeue-output.json -------------------------------------------------------------------------------- /torchx/schedulers/test/slurm_scheduler_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/slurm_scheduler_test.py -------------------------------------------------------------------------------- /torchx/schedulers/test/streams_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/streams_test.py -------------------------------------------------------------------------------- /torchx/schedulers/test/test_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/test_util.py -------------------------------------------------------------------------------- /torchx/schedulers/test/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/schedulers/test/train.py -------------------------------------------------------------------------------- /torchx/specs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/__init__.py -------------------------------------------------------------------------------- /torchx/specs/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/api.py -------------------------------------------------------------------------------- /torchx/specs/builders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/builders.py -------------------------------------------------------------------------------- /torchx/specs/file_linter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/file_linter.py -------------------------------------------------------------------------------- /torchx/specs/finder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/finder.py -------------------------------------------------------------------------------- /torchx/specs/named_resources_aws.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/named_resources_aws.py -------------------------------------------------------------------------------- /torchx/specs/named_resources_generic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/named_resources_generic.py -------------------------------------------------------------------------------- /torchx/specs/test/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/test/__init__.py -------------------------------------------------------------------------------- /torchx/specs/test/api_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/test/api_test.py -------------------------------------------------------------------------------- /torchx/specs/test/builders_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/test/builders_test.py -------------------------------------------------------------------------------- /torchx/specs/test/components/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/test/components/__init__.py -------------------------------------------------------------------------------- /torchx/specs/test/components/a/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/test/components/a/__init__.py -------------------------------------------------------------------------------- /torchx/specs/test/components/a/b/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/test/components/a/b/__init__.py -------------------------------------------------------------------------------- /torchx/specs/test/components/a/b/c.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/test/components/a/b/c.py -------------------------------------------------------------------------------- /torchx/specs/test/components/c/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/test/components/c/__init__.py -------------------------------------------------------------------------------- /torchx/specs/test/components/c/d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/test/components/c/d.py -------------------------------------------------------------------------------- /torchx/specs/test/file_linter_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/test/file_linter_test.py -------------------------------------------------------------------------------- /torchx/specs/test/finder_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/test/finder_test.py -------------------------------------------------------------------------------- /torchx/specs/test/named_resources_aws_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/test/named_resources_aws_test.py -------------------------------------------------------------------------------- /torchx/specs/test/named_resources_generic_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/test/named_resources_generic_test.py -------------------------------------------------------------------------------- /torchx/specs/test/named_resources_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/specs/test/named_resources_test.py -------------------------------------------------------------------------------- /torchx/test/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/test/__init__.py -------------------------------------------------------------------------------- /torchx/test/fixtures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/test/fixtures.py -------------------------------------------------------------------------------- /torchx/test/minio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/test/minio.py -------------------------------------------------------------------------------- /torchx/test/notebook_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/test/notebook_test.py -------------------------------------------------------------------------------- /torchx/test/suites.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/test/suites.py -------------------------------------------------------------------------------- /torchx/test/version_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/test/version_test.py -------------------------------------------------------------------------------- /torchx/tracker/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/tracker/__init__.py -------------------------------------------------------------------------------- /torchx/tracker/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/tracker/api.py -------------------------------------------------------------------------------- /torchx/tracker/backend/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/tracker/backend/__init__.py -------------------------------------------------------------------------------- /torchx/tracker/backend/fsspec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/tracker/backend/fsspec.py -------------------------------------------------------------------------------- /torchx/tracker/backend/test/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/tracker/backend/test/__init__.py -------------------------------------------------------------------------------- /torchx/tracker/backend/test/fsspec_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/tracker/backend/test/fsspec_test.py -------------------------------------------------------------------------------- /torchx/tracker/mlflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/tracker/mlflow.py -------------------------------------------------------------------------------- /torchx/tracker/test/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/tracker/test/__init__.py -------------------------------------------------------------------------------- /torchx/tracker/test/api_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/tracker/test/api_test.py -------------------------------------------------------------------------------- /torchx/tracker/test/mlflow_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/tracker/test/mlflow_test.py -------------------------------------------------------------------------------- /torchx/util/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchx/util/cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/util/cuda.py -------------------------------------------------------------------------------- /torchx/util/datetime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/util/datetime.py -------------------------------------------------------------------------------- /torchx/util/entrypoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/util/entrypoints.py -------------------------------------------------------------------------------- /torchx/util/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/util/io.py -------------------------------------------------------------------------------- /torchx/util/log_tee_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/util/log_tee_helpers.py -------------------------------------------------------------------------------- /torchx/util/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/util/modules.py -------------------------------------------------------------------------------- /torchx/util/session.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/util/session.py -------------------------------------------------------------------------------- /torchx/util/shlex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/util/shlex.py -------------------------------------------------------------------------------- /torchx/util/strings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/util/strings.py -------------------------------------------------------------------------------- /torchx/util/test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchx/util/test/cuda_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/util/test/cuda_test.py -------------------------------------------------------------------------------- /torchx/util/test/entrypoints_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/util/test/entrypoints_test.py -------------------------------------------------------------------------------- /torchx/util/test/modules_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/util/test/modules_test.py -------------------------------------------------------------------------------- /torchx/util/test/shlex_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/util/test/shlex_test.py -------------------------------------------------------------------------------- /torchx/util/test/strings_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/util/test/strings_test.py -------------------------------------------------------------------------------- /torchx/util/test/types_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/util/test/types_test.py -------------------------------------------------------------------------------- /torchx/util/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/util/types.py -------------------------------------------------------------------------------- /torchx/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/version.py -------------------------------------------------------------------------------- /torchx/workspace/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/workspace/__init__.py -------------------------------------------------------------------------------- /torchx/workspace/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/workspace/api.py -------------------------------------------------------------------------------- /torchx/workspace/dir_workspace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/workspace/dir_workspace.py -------------------------------------------------------------------------------- /torchx/workspace/docker_workspace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/workspace/docker_workspace.py -------------------------------------------------------------------------------- /torchx/workspace/test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchx/workspace/test/api_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/workspace/test/api_test.py -------------------------------------------------------------------------------- /torchx/workspace/test/dir_workspace_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/workspace/test/dir_workspace_test.py -------------------------------------------------------------------------------- /torchx/workspace/test/docker_workspace_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/workspace/test/docker_workspace_test.py -------------------------------------------------------------------------------- /torchx/workspace/test/expected_print_push_events.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/workspace/test/expected_print_push_events.txt -------------------------------------------------------------------------------- /torchx/workspace/test/mock_docker_push_events.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/torchx/workspace/test/mock_docker_push_events.json -------------------------------------------------------------------------------- /typestubs/torch/monitor.pyi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meta-pytorch/torchx/HEAD/typestubs/torch/monitor.pyi --------------------------------------------------------------------------------