├── .coveragerc ├── .dockerignore ├── .github └── workflows │ ├── build-docs.yml │ ├── build-test-publish-wheel.yml │ ├── changelog-build.yml │ ├── cherry-pick-release-commit.yml │ ├── close-inactive-issue-pr.yml │ ├── codeql.yml │ ├── community-bot.yml │ ├── config │ ├── changelog-config.json │ ├── codeql.yml │ └── typos.toml │ ├── copyright-check.yml │ ├── install.yml │ ├── nightly.yml │ ├── release-freeze.yml │ ├── release.yml │ ├── ruff-format.yml │ ├── ruff-lint.yml │ ├── spelling.yml │ └── test.yml ├── .gitignore ├── .pre-commit-config.yaml ├── .python-version ├── .vscode └── settings.json ├── CHANGELOG.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── codecov.yml ├── docs ├── conf.py ├── documentation.md ├── faqs.md ├── guides │ ├── cli.md │ ├── configuration.md │ ├── execution.md │ ├── index.md │ ├── management.md │ ├── ray.md │ └── why-use-nemo-run.md ├── index.md ├── project.json └── versions1.json ├── examples ├── dgxcloud │ └── sft-finetune │ │ ├── convert.sh │ │ ├── finetune-llama3.1-70b-sft.ipynb │ │ └── sft-finetune-llama3.1-70b.sh ├── docker │ └── hello_docker.py ├── entrypoint │ ├── README.md │ ├── experiment.py │ ├── img │ │ ├── experiment-2.png │ │ ├── experiment-3.png │ │ ├── experiment-4.png │ │ ├── experiment-5.png │ │ ├── experiment-6.png │ │ ├── experiment-help.png │ │ ├── task-2.png │ │ ├── task-3.png │ │ ├── task-4.png │ │ ├── task-5.png │ │ ├── task-6.png │ │ ├── task-7.png │ │ ├── task-help.png │ │ └── task-repl.gif │ ├── task.py │ ├── task.yaml │ ├── task_with_defaults.py │ └── test.toml └── hello-world │ ├── hello_experiments.ipynb │ ├── hello_scripts.py │ ├── hello_world.ipynb │ ├── scripts │ └── echo.sh │ └── simple │ ├── __init__.py │ └── add.py ├── nemo_run ├── __init__.py ├── __main__.py ├── api.py ├── cli │ ├── __init__.py │ ├── api.py │ ├── cli_parser.py │ ├── config.py │ ├── devspace.py │ ├── experiment.py │ ├── lazy.py │ └── workspace.py ├── config.py ├── core │ ├── __init__.py │ ├── execution │ │ ├── __init__.py │ │ ├── base.py │ │ ├── dgxcloud.py │ │ ├── docker.py │ │ ├── kuberay.py │ │ ├── launcher.py │ │ ├── lepton.py │ │ ├── local.py │ │ ├── skypilot.py │ │ ├── skypilot_jobs.py │ │ ├── slurm.py │ │ ├── templates │ │ │ ├── ft_launcher.j2 │ │ │ ├── slurm.sh.j2 │ │ │ └── slurm_ray.sh.j2 │ │ └── utils.py │ ├── frontend │ │ ├── __init__.py │ │ └── console │ │ │ ├── __init__.py │ │ │ ├── api.py │ │ │ └── styles.py │ ├── packaging │ │ ├── __init__.py │ │ ├── base.py │ │ ├── git.py │ │ ├── hybrid.py │ │ └── pattern.py │ ├── runners │ │ ├── __init__.py │ │ └── fdl_runner.py │ ├── serialization │ │ ├── __init__.py │ │ ├── yaml.py │ │ └── zlib_json.py │ └── tunnel │ │ ├── __init__.py │ │ ├── client.py │ │ ├── rsync.py │ │ └── server.py ├── devspace │ ├── __init__.py │ ├── base.py │ └── editor.py ├── exceptions.py ├── help.py ├── package_info.py └── run │ ├── __init__.py │ ├── api.py │ ├── experiment.py │ ├── job.py │ ├── logs.py │ ├── plugin.py │ ├── ray │ ├── __init__.py │ ├── cluster.py │ ├── job.py │ ├── kuberay.py │ ├── lepton.py │ ├── slurm.py │ └── templates │ │ ├── ray.sub.j2 │ │ └── ray_enroot.sub.j2 │ ├── task.py │ ├── torchx_backend │ ├── __init__.py │ ├── components │ │ ├── __init__.py │ │ ├── ft_launcher.py │ │ └── torchrun.py │ ├── launcher.py │ ├── packaging.py │ ├── runner.py │ └── schedulers │ │ ├── __init__.py │ │ ├── api.py │ │ ├── dgxcloud.py │ │ ├── docker.py │ │ ├── lepton.py │ │ ├── local.py │ │ ├── skypilot.py │ │ ├── skypilot_jobs.py │ │ ├── slurm.py │ │ └── utils.py │ └── utils.py ├── pyproject.toml ├── test ├── __init__.py ├── cli │ ├── __init__.py │ ├── test_api.py │ ├── test_cli_parser.py │ ├── test_config.py │ └── test_lazy.py ├── conftest.py ├── core │ ├── __init__.py │ ├── execution │ │ ├── __init__.py │ │ ├── artifacts │ │ │ ├── dummy_slurm.sh │ │ │ ├── expected_kuberay_cluster_advanced.yaml │ │ │ ├── expected_kuberay_cluster_basic.yaml │ │ │ ├── expected_kuberay_job_basic.yaml │ │ │ ├── expected_ray_cluster.sub │ │ │ ├── expected_ray_cluster_enroot.sub │ │ │ ├── expected_ray_cluster_ssh.sub │ │ │ ├── ft_het_slurm.sh │ │ │ ├── ft_slurm.sh │ │ │ ├── group_resource_req_slurm.sh │ │ │ ├── group_slurm.sh │ │ │ ├── group_slurm_no_monitor.sh │ │ │ └── het_slurm.sh │ │ ├── test_base.py │ │ ├── test_dgxcloud.py │ │ ├── test_docker.py │ │ ├── test_launcher.py │ │ ├── test_lepton.py │ │ ├── test_local.py │ │ ├── test_skypilot.py │ │ ├── test_skypilot_jobs.py │ │ ├── test_slurm.py │ │ ├── test_slurm_templates.py │ │ └── test_utils.py │ ├── frontend │ │ ├── __init__.py │ │ └── test_console.py │ ├── packaging │ │ ├── __init__.py │ │ ├── test_base.py │ │ ├── test_git.py │ │ ├── test_hybrid.py │ │ └── test_pattern.py │ └── tunnel │ │ ├── __init__.py │ │ ├── test_client.py │ │ ├── test_rsync.py │ │ └── test_server.py ├── devspace │ ├── __init__.py │ └── test_base.py ├── dummy_factory.py ├── dummy_script.py ├── dummy_type.py ├── run │ ├── ray │ │ ├── __init__.py │ │ ├── test_kuberay.py │ │ ├── test_lepton.py │ │ ├── test_slurm.py │ │ └── test_slurm_ray_request.py │ ├── test_experiment.py │ ├── test_job.py │ ├── test_logs.py │ ├── test_utils.py │ └── torchx_backend │ │ ├── __init__.py │ │ ├── schedulers │ │ ├── __init__.py │ │ ├── test_api.py │ │ ├── test_dgxcloud.py │ │ ├── test_docker.py │ │ ├── test_local.py │ │ ├── test_skypilot.py │ │ ├── test_skypilot_jobs.py │ │ └── test_slurm.py │ │ ├── test_api.py │ │ ├── test_launcher.py │ │ ├── test_packaging.py │ │ └── test_runner.py ├── test_api.py ├── test_config.py └── test_utils.py └── uv.lock /.coveragerc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.coveragerc -------------------------------------------------------------------------------- /.dockerignore: -------------------------------------------------------------------------------- 1 | # private space 2 | nogit/ 3 | .venv 4 | -------------------------------------------------------------------------------- /.github/workflows/build-docs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/build-docs.yml -------------------------------------------------------------------------------- /.github/workflows/build-test-publish-wheel.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/build-test-publish-wheel.yml -------------------------------------------------------------------------------- /.github/workflows/changelog-build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/changelog-build.yml -------------------------------------------------------------------------------- /.github/workflows/cherry-pick-release-commit.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/cherry-pick-release-commit.yml -------------------------------------------------------------------------------- /.github/workflows/close-inactive-issue-pr.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/close-inactive-issue-pr.yml -------------------------------------------------------------------------------- /.github/workflows/codeql.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/codeql.yml -------------------------------------------------------------------------------- /.github/workflows/community-bot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/community-bot.yml -------------------------------------------------------------------------------- /.github/workflows/config/changelog-config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/config/changelog-config.json -------------------------------------------------------------------------------- /.github/workflows/config/codeql.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/config/codeql.yml -------------------------------------------------------------------------------- /.github/workflows/config/typos.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/config/typos.toml -------------------------------------------------------------------------------- /.github/workflows/copyright-check.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/copyright-check.yml -------------------------------------------------------------------------------- /.github/workflows/install.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/install.yml -------------------------------------------------------------------------------- /.github/workflows/nightly.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/nightly.yml -------------------------------------------------------------------------------- /.github/workflows/release-freeze.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/release-freeze.yml -------------------------------------------------------------------------------- /.github/workflows/release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/release.yml -------------------------------------------------------------------------------- /.github/workflows/ruff-format.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/ruff-format.yml -------------------------------------------------------------------------------- /.github/workflows/ruff-lint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/ruff-lint.yml -------------------------------------------------------------------------------- /.github/workflows/spelling.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/spelling.yml -------------------------------------------------------------------------------- /.github/workflows/test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.github/workflows/test.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.python-version: -------------------------------------------------------------------------------- 1 | 3.11.9 2 | -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/.vscode/settings.json -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/README.md -------------------------------------------------------------------------------- /codecov.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/codecov.yml -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/documentation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/docs/documentation.md -------------------------------------------------------------------------------- /docs/faqs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/docs/faqs.md -------------------------------------------------------------------------------- /docs/guides/cli.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/docs/guides/cli.md -------------------------------------------------------------------------------- /docs/guides/configuration.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/docs/guides/configuration.md -------------------------------------------------------------------------------- /docs/guides/execution.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/docs/guides/execution.md -------------------------------------------------------------------------------- /docs/guides/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/docs/guides/index.md -------------------------------------------------------------------------------- /docs/guides/management.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/docs/guides/management.md -------------------------------------------------------------------------------- /docs/guides/ray.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/docs/guides/ray.md -------------------------------------------------------------------------------- /docs/guides/why-use-nemo-run.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/docs/guides/why-use-nemo-run.md -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/project.json: -------------------------------------------------------------------------------- 1 | {"name": "NeMo-Run", "version": "0.5.0"} 2 | -------------------------------------------------------------------------------- /docs/versions1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/docs/versions1.json -------------------------------------------------------------------------------- /examples/dgxcloud/sft-finetune/convert.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/dgxcloud/sft-finetune/convert.sh -------------------------------------------------------------------------------- /examples/dgxcloud/sft-finetune/finetune-llama3.1-70b-sft.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/dgxcloud/sft-finetune/finetune-llama3.1-70b-sft.ipynb -------------------------------------------------------------------------------- /examples/dgxcloud/sft-finetune/sft-finetune-llama3.1-70b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/dgxcloud/sft-finetune/sft-finetune-llama3.1-70b.sh -------------------------------------------------------------------------------- /examples/docker/hello_docker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/docker/hello_docker.py -------------------------------------------------------------------------------- /examples/entrypoint/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/README.md -------------------------------------------------------------------------------- /examples/entrypoint/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/experiment.py -------------------------------------------------------------------------------- /examples/entrypoint/img/experiment-2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/img/experiment-2.png -------------------------------------------------------------------------------- /examples/entrypoint/img/experiment-3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/img/experiment-3.png -------------------------------------------------------------------------------- /examples/entrypoint/img/experiment-4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/img/experiment-4.png -------------------------------------------------------------------------------- /examples/entrypoint/img/experiment-5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/img/experiment-5.png -------------------------------------------------------------------------------- /examples/entrypoint/img/experiment-6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/img/experiment-6.png -------------------------------------------------------------------------------- /examples/entrypoint/img/experiment-help.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/img/experiment-help.png -------------------------------------------------------------------------------- /examples/entrypoint/img/task-2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/img/task-2.png -------------------------------------------------------------------------------- /examples/entrypoint/img/task-3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/img/task-3.png -------------------------------------------------------------------------------- /examples/entrypoint/img/task-4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/img/task-4.png -------------------------------------------------------------------------------- /examples/entrypoint/img/task-5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/img/task-5.png -------------------------------------------------------------------------------- /examples/entrypoint/img/task-6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/img/task-6.png -------------------------------------------------------------------------------- /examples/entrypoint/img/task-7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/img/task-7.png -------------------------------------------------------------------------------- /examples/entrypoint/img/task-help.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/img/task-help.png -------------------------------------------------------------------------------- /examples/entrypoint/img/task-repl.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/img/task-repl.gif -------------------------------------------------------------------------------- /examples/entrypoint/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/task.py -------------------------------------------------------------------------------- /examples/entrypoint/task.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/task.yaml -------------------------------------------------------------------------------- /examples/entrypoint/task_with_defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/task_with_defaults.py -------------------------------------------------------------------------------- /examples/entrypoint/test.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/entrypoint/test.toml -------------------------------------------------------------------------------- /examples/hello-world/hello_experiments.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/hello-world/hello_experiments.ipynb -------------------------------------------------------------------------------- /examples/hello-world/hello_scripts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/hello-world/hello_scripts.py -------------------------------------------------------------------------------- /examples/hello-world/hello_world.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/hello-world/hello_world.ipynb -------------------------------------------------------------------------------- /examples/hello-world/scripts/echo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/hello-world/scripts/echo.sh -------------------------------------------------------------------------------- /examples/hello-world/simple/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/hello-world/simple/__init__.py -------------------------------------------------------------------------------- /examples/hello-world/simple/add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/examples/hello-world/simple/add.py -------------------------------------------------------------------------------- /nemo_run/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/__init__.py -------------------------------------------------------------------------------- /nemo_run/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/__main__.py -------------------------------------------------------------------------------- /nemo_run/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/api.py -------------------------------------------------------------------------------- /nemo_run/cli/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/cli/__init__.py -------------------------------------------------------------------------------- /nemo_run/cli/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/cli/api.py -------------------------------------------------------------------------------- /nemo_run/cli/cli_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/cli/cli_parser.py -------------------------------------------------------------------------------- /nemo_run/cli/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/cli/config.py -------------------------------------------------------------------------------- /nemo_run/cli/devspace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/cli/devspace.py -------------------------------------------------------------------------------- /nemo_run/cli/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/cli/experiment.py -------------------------------------------------------------------------------- /nemo_run/cli/lazy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/cli/lazy.py -------------------------------------------------------------------------------- /nemo_run/cli/workspace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/cli/workspace.py -------------------------------------------------------------------------------- /nemo_run/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/config.py -------------------------------------------------------------------------------- /nemo_run/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/__init__.py -------------------------------------------------------------------------------- /nemo_run/core/execution/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/execution/__init__.py -------------------------------------------------------------------------------- /nemo_run/core/execution/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/execution/base.py -------------------------------------------------------------------------------- /nemo_run/core/execution/dgxcloud.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/execution/dgxcloud.py -------------------------------------------------------------------------------- /nemo_run/core/execution/docker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/execution/docker.py -------------------------------------------------------------------------------- /nemo_run/core/execution/kuberay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/execution/kuberay.py -------------------------------------------------------------------------------- /nemo_run/core/execution/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/execution/launcher.py -------------------------------------------------------------------------------- /nemo_run/core/execution/lepton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/execution/lepton.py -------------------------------------------------------------------------------- /nemo_run/core/execution/local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/execution/local.py -------------------------------------------------------------------------------- /nemo_run/core/execution/skypilot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/execution/skypilot.py -------------------------------------------------------------------------------- /nemo_run/core/execution/skypilot_jobs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/execution/skypilot_jobs.py -------------------------------------------------------------------------------- /nemo_run/core/execution/slurm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/execution/slurm.py -------------------------------------------------------------------------------- /nemo_run/core/execution/templates/ft_launcher.j2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/execution/templates/ft_launcher.j2 -------------------------------------------------------------------------------- /nemo_run/core/execution/templates/slurm.sh.j2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/execution/templates/slurm.sh.j2 -------------------------------------------------------------------------------- /nemo_run/core/execution/templates/slurm_ray.sh.j2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/execution/templates/slurm_ray.sh.j2 -------------------------------------------------------------------------------- /nemo_run/core/execution/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/execution/utils.py -------------------------------------------------------------------------------- /nemo_run/core/frontend/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/frontend/__init__.py -------------------------------------------------------------------------------- /nemo_run/core/frontend/console/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/frontend/console/__init__.py -------------------------------------------------------------------------------- /nemo_run/core/frontend/console/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/frontend/console/api.py -------------------------------------------------------------------------------- /nemo_run/core/frontend/console/styles.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/frontend/console/styles.py -------------------------------------------------------------------------------- /nemo_run/core/packaging/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/packaging/__init__.py -------------------------------------------------------------------------------- /nemo_run/core/packaging/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/packaging/base.py -------------------------------------------------------------------------------- /nemo_run/core/packaging/git.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/packaging/git.py -------------------------------------------------------------------------------- /nemo_run/core/packaging/hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/packaging/hybrid.py -------------------------------------------------------------------------------- /nemo_run/core/packaging/pattern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/packaging/pattern.py -------------------------------------------------------------------------------- /nemo_run/core/runners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/runners/__init__.py -------------------------------------------------------------------------------- /nemo_run/core/runners/fdl_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/runners/fdl_runner.py -------------------------------------------------------------------------------- /nemo_run/core/serialization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/serialization/__init__.py -------------------------------------------------------------------------------- /nemo_run/core/serialization/yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/serialization/yaml.py -------------------------------------------------------------------------------- /nemo_run/core/serialization/zlib_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/serialization/zlib_json.py -------------------------------------------------------------------------------- /nemo_run/core/tunnel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/tunnel/__init__.py -------------------------------------------------------------------------------- /nemo_run/core/tunnel/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/tunnel/client.py -------------------------------------------------------------------------------- /nemo_run/core/tunnel/rsync.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/tunnel/rsync.py -------------------------------------------------------------------------------- /nemo_run/core/tunnel/server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/core/tunnel/server.py -------------------------------------------------------------------------------- /nemo_run/devspace/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/devspace/__init__.py -------------------------------------------------------------------------------- /nemo_run/devspace/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/devspace/base.py -------------------------------------------------------------------------------- /nemo_run/devspace/editor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/devspace/editor.py -------------------------------------------------------------------------------- /nemo_run/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/exceptions.py -------------------------------------------------------------------------------- /nemo_run/help.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/help.py -------------------------------------------------------------------------------- /nemo_run/package_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/package_info.py -------------------------------------------------------------------------------- /nemo_run/run/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/__init__.py -------------------------------------------------------------------------------- /nemo_run/run/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/api.py -------------------------------------------------------------------------------- /nemo_run/run/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/experiment.py -------------------------------------------------------------------------------- /nemo_run/run/job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/job.py -------------------------------------------------------------------------------- /nemo_run/run/logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/logs.py -------------------------------------------------------------------------------- /nemo_run/run/plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/plugin.py -------------------------------------------------------------------------------- /nemo_run/run/ray/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /nemo_run/run/ray/cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/ray/cluster.py -------------------------------------------------------------------------------- /nemo_run/run/ray/job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/ray/job.py -------------------------------------------------------------------------------- /nemo_run/run/ray/kuberay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/ray/kuberay.py -------------------------------------------------------------------------------- /nemo_run/run/ray/lepton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/ray/lepton.py -------------------------------------------------------------------------------- /nemo_run/run/ray/slurm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/ray/slurm.py -------------------------------------------------------------------------------- /nemo_run/run/ray/templates/ray.sub.j2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/ray/templates/ray.sub.j2 -------------------------------------------------------------------------------- /nemo_run/run/ray/templates/ray_enroot.sub.j2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/ray/templates/ray_enroot.sub.j2 -------------------------------------------------------------------------------- /nemo_run/run/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/task.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/torchx_backend/__init__.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/components/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/torchx_backend/components/__init__.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/components/ft_launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/torchx_backend/components/ft_launcher.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/components/torchrun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/torchx_backend/components/torchrun.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/torchx_backend/launcher.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/packaging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/torchx_backend/packaging.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/torchx_backend/runner.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/schedulers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/torchx_backend/schedulers/__init__.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/schedulers/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/torchx_backend/schedulers/api.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/schedulers/dgxcloud.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/torchx_backend/schedulers/dgxcloud.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/schedulers/docker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/torchx_backend/schedulers/docker.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/schedulers/lepton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/torchx_backend/schedulers/lepton.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/schedulers/local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/torchx_backend/schedulers/local.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/schedulers/skypilot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/torchx_backend/schedulers/skypilot.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/schedulers/skypilot_jobs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/torchx_backend/schedulers/skypilot_jobs.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/schedulers/slurm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/torchx_backend/schedulers/slurm.py -------------------------------------------------------------------------------- /nemo_run/run/torchx_backend/schedulers/utils.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /nemo_run/run/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/nemo_run/run/utils.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/pyproject.toml -------------------------------------------------------------------------------- /test/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/__init__.py -------------------------------------------------------------------------------- /test/cli/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/cli/test_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/cli/test_api.py -------------------------------------------------------------------------------- /test/cli/test_cli_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/cli/test_cli_parser.py -------------------------------------------------------------------------------- /test/cli/test_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/cli/test_config.py -------------------------------------------------------------------------------- /test/cli/test_lazy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/cli/test_lazy.py -------------------------------------------------------------------------------- /test/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/conftest.py -------------------------------------------------------------------------------- /test/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/__init__.py -------------------------------------------------------------------------------- /test/core/execution/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/__init__.py -------------------------------------------------------------------------------- /test/core/execution/artifacts/dummy_slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/artifacts/dummy_slurm.sh -------------------------------------------------------------------------------- /test/core/execution/artifacts/expected_kuberay_cluster_advanced.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/artifacts/expected_kuberay_cluster_advanced.yaml -------------------------------------------------------------------------------- /test/core/execution/artifacts/expected_kuberay_cluster_basic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/artifacts/expected_kuberay_cluster_basic.yaml -------------------------------------------------------------------------------- /test/core/execution/artifacts/expected_kuberay_job_basic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/artifacts/expected_kuberay_job_basic.yaml -------------------------------------------------------------------------------- /test/core/execution/artifacts/expected_ray_cluster.sub: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/artifacts/expected_ray_cluster.sub -------------------------------------------------------------------------------- /test/core/execution/artifacts/expected_ray_cluster_enroot.sub: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/artifacts/expected_ray_cluster_enroot.sub -------------------------------------------------------------------------------- /test/core/execution/artifacts/expected_ray_cluster_ssh.sub: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/artifacts/expected_ray_cluster_ssh.sub -------------------------------------------------------------------------------- /test/core/execution/artifacts/ft_het_slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/artifacts/ft_het_slurm.sh -------------------------------------------------------------------------------- /test/core/execution/artifacts/ft_slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/artifacts/ft_slurm.sh -------------------------------------------------------------------------------- /test/core/execution/artifacts/group_resource_req_slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/artifacts/group_resource_req_slurm.sh -------------------------------------------------------------------------------- /test/core/execution/artifacts/group_slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/artifacts/group_slurm.sh -------------------------------------------------------------------------------- /test/core/execution/artifacts/group_slurm_no_monitor.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/artifacts/group_slurm_no_monitor.sh -------------------------------------------------------------------------------- /test/core/execution/artifacts/het_slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/artifacts/het_slurm.sh -------------------------------------------------------------------------------- /test/core/execution/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/test_base.py -------------------------------------------------------------------------------- /test/core/execution/test_dgxcloud.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/test_dgxcloud.py -------------------------------------------------------------------------------- /test/core/execution/test_docker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/test_docker.py -------------------------------------------------------------------------------- /test/core/execution/test_launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/test_launcher.py -------------------------------------------------------------------------------- /test/core/execution/test_lepton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/test_lepton.py -------------------------------------------------------------------------------- /test/core/execution/test_local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/test_local.py -------------------------------------------------------------------------------- /test/core/execution/test_skypilot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/test_skypilot.py -------------------------------------------------------------------------------- /test/core/execution/test_skypilot_jobs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/test_skypilot_jobs.py -------------------------------------------------------------------------------- /test/core/execution/test_slurm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/test_slurm.py -------------------------------------------------------------------------------- /test/core/execution/test_slurm_templates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/test_slurm_templates.py -------------------------------------------------------------------------------- /test/core/execution/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/execution/test_utils.py -------------------------------------------------------------------------------- /test/core/frontend/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/frontend/__init__.py -------------------------------------------------------------------------------- /test/core/frontend/test_console.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/frontend/test_console.py -------------------------------------------------------------------------------- /test/core/packaging/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/packaging/__init__.py -------------------------------------------------------------------------------- /test/core/packaging/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/packaging/test_base.py -------------------------------------------------------------------------------- /test/core/packaging/test_git.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/packaging/test_git.py -------------------------------------------------------------------------------- /test/core/packaging/test_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/packaging/test_hybrid.py -------------------------------------------------------------------------------- /test/core/packaging/test_pattern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/packaging/test_pattern.py -------------------------------------------------------------------------------- /test/core/tunnel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/tunnel/__init__.py -------------------------------------------------------------------------------- /test/core/tunnel/test_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/tunnel/test_client.py -------------------------------------------------------------------------------- /test/core/tunnel/test_rsync.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/tunnel/test_rsync.py -------------------------------------------------------------------------------- /test/core/tunnel/test_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/core/tunnel/test_server.py -------------------------------------------------------------------------------- /test/devspace/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/devspace/__init__.py -------------------------------------------------------------------------------- /test/devspace/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/devspace/test_base.py -------------------------------------------------------------------------------- /test/dummy_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/dummy_factory.py -------------------------------------------------------------------------------- /test/dummy_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/dummy_script.py -------------------------------------------------------------------------------- /test/dummy_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/dummy_type.py -------------------------------------------------------------------------------- /test/run/ray/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/run/ray/test_kuberay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/ray/test_kuberay.py -------------------------------------------------------------------------------- /test/run/ray/test_lepton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/ray/test_lepton.py -------------------------------------------------------------------------------- /test/run/ray/test_slurm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/ray/test_slurm.py -------------------------------------------------------------------------------- /test/run/ray/test_slurm_ray_request.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/ray/test_slurm_ray_request.py -------------------------------------------------------------------------------- /test/run/test_experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/test_experiment.py -------------------------------------------------------------------------------- /test/run/test_job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/test_job.py -------------------------------------------------------------------------------- /test/run/test_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/test_logs.py -------------------------------------------------------------------------------- /test/run/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/test_utils.py -------------------------------------------------------------------------------- /test/run/torchx_backend/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/torchx_backend/__init__.py -------------------------------------------------------------------------------- /test/run/torchx_backend/schedulers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/torchx_backend/schedulers/__init__.py -------------------------------------------------------------------------------- /test/run/torchx_backend/schedulers/test_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/torchx_backend/schedulers/test_api.py -------------------------------------------------------------------------------- /test/run/torchx_backend/schedulers/test_dgxcloud.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/torchx_backend/schedulers/test_dgxcloud.py -------------------------------------------------------------------------------- /test/run/torchx_backend/schedulers/test_docker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/torchx_backend/schedulers/test_docker.py -------------------------------------------------------------------------------- /test/run/torchx_backend/schedulers/test_local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/torchx_backend/schedulers/test_local.py -------------------------------------------------------------------------------- /test/run/torchx_backend/schedulers/test_skypilot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/torchx_backend/schedulers/test_skypilot.py -------------------------------------------------------------------------------- /test/run/torchx_backend/schedulers/test_skypilot_jobs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/torchx_backend/schedulers/test_skypilot_jobs.py -------------------------------------------------------------------------------- /test/run/torchx_backend/schedulers/test_slurm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/torchx_backend/schedulers/test_slurm.py -------------------------------------------------------------------------------- /test/run/torchx_backend/test_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/torchx_backend/test_api.py -------------------------------------------------------------------------------- /test/run/torchx_backend/test_launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/torchx_backend/test_launcher.py -------------------------------------------------------------------------------- /test/run/torchx_backend/test_packaging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/torchx_backend/test_packaging.py -------------------------------------------------------------------------------- /test/run/torchx_backend/test_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/run/torchx_backend/test_runner.py -------------------------------------------------------------------------------- /test/test_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/test_api.py -------------------------------------------------------------------------------- /test/test_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/test_config.py -------------------------------------------------------------------------------- /test/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/test/test_utils.py -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-NeMo/Run/HEAD/uv.lock --------------------------------------------------------------------------------