pytorch-ignite 0.6.0.dev20250612__tar.gz → 0.6.0.dev20251124__tar.gz

pytorch_ignite-0.6.0.dev20251124/.git-blame-ignore-revs

@@ -0,0 +1,7 @@
+# Run this command to always ignore formatting commits in `git blame`
+# git config blame.ignoreRevsFile .git-blame-ignore-revs
+
+# Run pre-commit to fix end-of-file-fixer and trailing-whitespace (#3418)
+fe5a29b6942829ff2870181b6c1a68ebd2af9add
+# Fix code style (#3423)
+f9e8ca106a0d3f386f26f3a9481fd2eb341999f0

pytorch_ignite-0.6.0.dev20251124/.github/FUNDING.yml

@@ -0,0 +1,12 @@
+# These are supported funding model platforms
+
+github: [vfdev-5] # Replace with up to 4 GitHub Sponsors-enabled usernames e.g., [user1, user2]
+patreon: # Replace with a single Patreon username
+open_collective: pytorch-ignite # Replace with a single Open Collective username
+ko_fi: # Replace with a single Ko-fi username
+tidelift: # Replace with a single Tidelift platform-name/package-name e.g., npm/babel
+community_bridge: # Replace with a single Community Bridge project-name e.g., cloud-foundry
+liberapay: # Replace with a single Liberapay username
+issuehunt: # Replace with a single IssueHunt username
+otechie: # Replace with a single Otechie username
+custom: # Replace with up to 4 custom sponsorship URLs e.g., ['link1', 'link2']

pytorch_ignite-0.6.0.dev20251124/.github/ISSUE_TEMPLATE/bug-report.md

@@ -0,0 +1,20 @@
+---
+name: "\U0001F41B Bug Report"
+about: Submit a bug report to help us improve Ignite
+---
+
+## 🐛 Bug description
+
+<!-- A clear and concise description of what the bug is. -->
+<!-- Please, add steps on how to reproduce it. -->
+<!-- If you have a code sample, error messages, stack traces, please provide it here as well -->
+<!-- A clear and concise description of what you expected to happen. -->
+
+## Environment
+
+- PyTorch Version (e.g., 1.4):
+- Ignite Version (e.g., 0.3.0):
+- OS (e.g., Linux):
+- How you installed Ignite (`conda`, `pip`, source):
+- Python version:
+- Any other relevant information:

pytorch_ignite-0.6.0.dev20251124/.github/ISSUE_TEMPLATE/documentation.md

@@ -0,0 +1,9 @@
+---
+name: "\U0001F4DA Documentation"
+about: Report an issue, comment or suggestion related to project's docs
+labels: "docs"
+---
+
+## 📚 Documentation
+
+<!-- Describe your issue, comment or suggestion related to Ignite's docs: https://pytorch.org/ignite/, CONTRIBUTING.md and other documentations. Thank you! -->

pytorch_ignite-0.6.0.dev20251124/.github/ISSUE_TEMPLATE/feature-request.md

@@ -0,0 +1,12 @@
+---
+name: "\U0001F680 Feature Request"
+about: Submit a proposal/request for a new Ingite feature
+---
+
+## 🚀 Feature
+
+<!-- A clear and concise description of the feature proposal -->
+<!-- Please outline the motivation for the proposal. Is your feature request related to a problem? e.g., I'm always frustrated when [...]. If this is related to another GitHub issue, please link here too -->
+<!-- A clear and concise description of what you want to happen. -->
+<!-- A clear and concise description of any alternative solutions or features you've considered, if any. -->
+<!-- Add any other context or screenshots about the feature request here. -->

pytorch_ignite-0.6.0.dev20251124/.github/ISSUE_TEMPLATE/questions-help-support.md

@@ -0,0 +1,11 @@
+---
+name: "❓Questions/Help/Support"
+about: Do you have a question?
+labels: "question"
+---
+
+## ❓ Questions/Help/Support
+
+<!-- If you have a question, please checkout other issues labelled as "question" (https://github.com/pytorch/ignite/issues?q=is%3Aissue+is%3Aopen+label%3Aquestion) -->
+<!-- If your question is not listed above or you need a support or help, please provide a description-->
+<!-- If you prefer, you can also reach out us on [Discussion Forum](https://discuss.pytorch.org/) adding tag "ignite". -->

pytorch_ignite-0.6.0.dev20251124/.github/ISSUE_TEMPLATE/user-feedback.md

@@ -0,0 +1,12 @@
+---
+name: "\U0001F44D User feedback"
+about: Say thanks or why you don't like
+title: ""
+labels: ""
+assignees: ""
+---
+
+> This is a place to leave any feedback on this package.
+> If you like the work, feel free to say thanks here
+> If you do not like something, please, share it with us and we can see how to improve
+> Thank you !

pytorch_ignite-0.6.0.dev20251124/.github/PULL_REQUEST_TEMPLATE.md

@@ -0,0 +1,9 @@
+Fixes #{issue number}
+
+Description:
+
+Check list:
+
+- [ ] New tests are added (if a new feature is added)
+- [ ] New doc strings: description and/or example code are in RST format
+- [ ] Documentation is updated (if required)

pytorch_ignite-0.6.0.dev20251124/.github/dependabot.yml

@@ -0,0 +1,10 @@
+version: 2
+updates:
+  - package-ecosystem: "github-actions"
+    directory: "/"
+    schedule:
+      interval: "monthly"
+    groups:
+      github-actions:
+        patterns:
+          - "*"

pytorch_ignite-0.6.0.dev20251124/.github/failed_schedule_issue_template.md

@@ -0,0 +1,12 @@
+---
+title: Scheduled workflow failed
+labels:
+  - bug
+---
+
+Oh no, something went wrong in the scheduled workflow **{{ env.GITHUB_WORKFLOW }} with commit {{ env.GITHUB_SHA }}**.
+Please look into it:
+
+{{ env.GITHUB_SERVER_URL }}/{{ env.GITHUB_REPOSITORY }}/actions/runs/{{ env.GITHUB_RUN_ID }}
+
+Feel free to close this if this was just a one-off error.

pytorch_ignite-0.6.0.dev20251124/.github/pr-labeler-config.yml

@@ -0,0 +1,55 @@
+# Add 'docker' to any changes within 'docker' folder or any subfolders
+docker:
+  - changed-files:
+      - any-glob-to-any-file: docker/**
+
+# Add 'docs' to any changes within 'docs' folder
+docs:
+  - changed-files:
+      - any-glob-to-any-file: docs/**
+
+# Add 'ci' to any changes in '.github' folder
+ci:
+  - changed-files:
+      - any-glob-to-any-file: .github/**
+
+# Add 'examples' to any changes within 'examples' folder
+examples:
+  - changed-files:
+      - any-glob-to-any-file: examples/**
+
+# Add 'base' to any changes within 'base' folder
+"module: base":
+  - changed-files:
+      - any-glob-to-any-file: ignite/base/**/*
+
+# Add 'contrib' to any changes within 'contrib' folder
+"module: contrib":
+  - changed-files:
+      - any-glob-to-any-file: ignite/contrib/**/*
+
+# Add 'distributed' to any changes within 'distributed' folder
+"module: distributed":
+  - changed-files:
+      - any-glob-to-any-file: ignite/distributed/**/*
+
+# Add 'engine' to any changes within 'engine' folder
+"module: engine":
+  - changed-files:
+      - any-glob-to-any-file: ignite/engine/**/*
+
+# Add 'handlers' to any changes within 'handlers' folder
+"module: handlers":
+  - changed-files:
+      - any-glob-to-any-file: ignite/handlers/**/*
+
+# Add 'metrics' to any changes within 'metrics' folder
+"module: metrics":
+  - changed-files:
+      - any-glob-to-any-file: ignite/metrics/**/*
+
+  -
+# Add 'utils' to any changes within 'utils' module
+"module: utils":
+  - changed-files:
+      - any-glob-to-any-file: ignite/utils.py

pytorch_ignite-0.6.0.dev20251124/.github/workflows/binaries-nightly-release.yml

@@ -0,0 +1,62 @@
+name: Nightly Releases
+
+on:
+  # https://docs.github.com/en/free-pro-team@latest/actions/reference/workflow-syntax-for-github-actions#onschedule
+  schedule:
+    # Run at 00:00 UTC Every Day
+    - cron: "0 0 * * *"
+
+jobs:
+  build-publish:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v5
+
+      - name: Setup Miniconda
+        uses: conda-incubator/setup-miniconda@v3
+        with:
+          miniconda-version: "latest"
+          python-version: "3.10"
+          channels: defaults,pytorch-nightly
+
+      - name: Setup nightly version
+        run: |
+          sed -i "s/__version__ = \"\(.*\)\"/__version__ = \"\1.dev$(date -u +%Y%m%d)\"/g" ignite/__init__.py
+          cat ignite/__init__.py
+          sed -i "s/__version__ = \"\(.*\)\"/__version__ = \"\1.dev$(date -u +%Y%m%d)\"/g" conda.recipe/meta.yaml
+          cat conda.recipe/meta.yaml
+
+      - name: Install dependencies
+        shell: bash -l {0}
+        run: |
+          conda install -y pytorch torchvision cpuonly -c pytorch-nightly
+          pip install -r requirements-dev.txt
+
+      - name: Build and Publish Conda binaries
+        shell: bash -l {0}
+        env:
+          ANACONDA_TOKEN: ${{ secrets.ANACONDA_TOKEN }}
+          UPLOAD_USER: "pytorch-nightly"
+        run: |
+          # Update version
+          chmod +x ./conda.recipe/build_and_upload.sh
+          ./conda.recipe/build_and_upload.sh
+
+      - name: Build and Publish PyPI binaries
+        shell: bash -l {0}
+        run: |
+          # workaround to fix https://github.com/pytorch/ignite/issues/2373
+          pip uninstall -y twine pkginfo
+          pip install --upgrade --no-cache-dir hatch twine 'pkginfo>=1.8.2'
+          hatch build
+          twine --version
+          twine check dist/*
+          TWINE_USERNAME="${{ secrets.PYPI_USER }}" TWINE_PASSWORD="${{ secrets.PYPI_TOKEN }}" twine upload --verbose dist/*
+
+      - uses: JasonEtco/create-an-issue@v2
+        name: Create issue if nightly releases failed
+        if: failure()
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        with:
+          filename: .github/failed_schedule_issue_template.md

pytorch_ignite-0.6.0.dev20251124/.github/workflows/build_docs.sh

@@ -0,0 +1 @@
+sphinx-versioning --use-master-conf --use-master-templates build --greatest-tag --whitelist-branches master docs/source docs/build/html

pytorch_ignite-0.6.0.dev20251124/.github/workflows/code-style-checks.yml

@@ -0,0 +1,65 @@
+name: Code Style Checks
+
+on:
+  push:
+    branches:
+      - master
+      - "*.*.*"
+    paths:
+      - "**.py"
+      - "setup.cfg"
+      - "requirements-dev.txt"
+      - "pyproject.toml"
+      - ".pre-commit-config.yaml"
+      - ".github/workflows/code-style-checks.yml"
+      - "!assets/**"
+      - "!docker/**"
+      - "!docs/**"
+      - "!conda.recipe"
+  pull_request:
+    paths:
+      - "**.py"
+      - "setup.cfg"
+      - "requirements-dev.txt"
+      - "pyproject.toml"
+      - ".pre-commit-config.yaml"
+      - ".github/workflows/code-style-checks.yml"
+      - "!assets/**"
+      - "!docker/**"
+      - "!docs/**"
+      - "!conda.recipe"
+  workflow_dispatch:
+
+concurrency:
+  # <workflow_name>-<branch_name>-<true || commit_sha (if branch is protected)>
+  group: code-style-${{ github.ref_name }}-${{ !(github.ref_protected) || github.sha }}
+  cancel-in-progress: true
+
+jobs:
+  code-style:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.10", "3.13"]
+
+    steps:
+      - uses: actions/checkout@v5
+
+      - uses: astral-sh/setup-uv@v7
+        with:
+          version: "latest"
+          python-version: ${{ matrix.python-version }}
+          activate-environment: true
+          enable-cache: true
+          cache-dependency-glob: |
+            **/requirements-dev.txt
+            **/pyproject.toml
+
+      - name: Install dependencies
+        run: |
+          uv pip install pre-commit
+          uv pip install -r requirements-dev.txt
+          uv pip install -e .
+
+      - name: Run pre-commit checks
+        run: pre-commit run --all-files --show-diff-on-failure

pytorch_ignite-0.6.0.dev20251124/.github/workflows/discord_issues.yml

@@ -0,0 +1,28 @@
+name: Discuss "help-wanted" issue on Discord
+
+on:
+  issues:
+    types:
+      - labeled
+  workflow_dispatch:
+    inputs:
+      issue_number:
+        description: "Issue number"
+        required: true
+
+permissions:
+  issues: write
+
+jobs:
+  discord:
+    runs-on: ubuntu-latest
+    steps:
+      - name: "Discuss on Discord-Issues"
+        if: ${{ github.event.label.name == 'help wanted' }}
+        uses: EndBug/discuss-on-discord@v1.1.0
+        with:
+          discord_bot_token: ${{ secrets.DISCORD_BOT_TOKEN }}
+          destination: ${{ secrets.DISCORD_BOT_DESTINATION }}
+          issue_number: ${{ github.event.inputs.issue_number || github.event.issue.number }}
+          issue_comment: Hey 👋, I've just created a [thread]($THREAD_LINK$) for this issue on [PyTorch-Ignite Discord](https://pytorch-ignite.ai/chat) where you can quickly talk to the community on the topic.
+          discord_message: New issue created in `${{ github.repository }}`:<https://github.com/${{ github.repository }}/issues/${{ github.event.issue.number }}>

pytorch_ignite-0.6.0.dev20251124/.github/workflows/discord_pull_requests.yaml

@@ -0,0 +1,28 @@
+name: Discuss "help-wanted" PR on Discord
+
+on:
+  pull_request:
+    types:
+      - labeled
+  workflow_dispatch:
+    inputs:
+      pull_request_number:
+        description: "Pull request number"
+        required: true
+
+permissions:
+  pull-requests: write
+
+jobs:
+  discord:
+    runs-on: ubuntu-latest
+    steps:
+      - name: "Discuss on Discord-PR (Non-maintainer only)"
+        if: ${{ github.event.label.name == 'help wanted' }}
+        uses: EndBug/discuss-on-discord@v1.1.0
+        with:
+          discord_bot_token: ${{ secrets.DISCORD_BOT_TOKEN }}
+          destination: ${{ secrets.DISCORD_BOT_DESTINATION }}
+          issue_number: ${{ github.event.inputs.pull_request_number || github.event.pull_request.number }}
+          issue_comment: Hey 👋, I've just created a [thread]($THREAD_LINK$) for this pull request on [PyTorch-Ignite Discord](https://pytorch-ignite.ai/chat) where you can quickly talk to the community on the topic.
+          discord_message: New PR created in `${{ github.repository }}`:<https://github.com/${{ github.repository }}/pull/${{ github.event.pull_request_number || github.event.pull_request.number }}>

pytorch_ignite-0.6.0.dev20251124/.github/workflows/docker-build.yml

@@ -0,0 +1,281 @@
+name: Build/Publish Docker Images
+
+on:
+  pull_request:
+    paths:
+      - docker/**
+      - ".github/workflows/docker-build.yml"
+  release:
+    types: [published]
+  workflow_dispatch:
+
+concurrency:
+  # <workflow_name>-<branch_name>-<true || commit_sha (if branch is protected)>
+  group: docker-build-publish-${{ github.ref_name }}-${{ !(github.ref_protected) || github.sha }}
+  cancel-in-progress: true
+
+jobs:
+  setup:
+    name: Setup
+    runs-on: ubuntu-latest
+    outputs:
+      modified: ${{ steps.set-modified.outputs.modified }}
+      pth_version: ${{ steps.set-versions.outputs.pth_version }}
+      hvd_version: ${{ steps.set-versions.outputs.hvd_version }}
+      msdp_version: ${{ steps.set-versions.outputs.msdp_version }}
+    steps:
+      - uses: actions/checkout@v5
+      - name: Changed Files Exporter
+        if: github.event_name == 'pull_request'
+        id: files
+        uses: umani/changed-files@v4.2.0
+        with:
+          repo-token: ${{ secrets.GITHUB_TOKEN }}
+      - name: Get a list of modified files
+        if: github.event_name == 'pull_request'
+        run: echo "modified=${{ steps.files.outputs.files_created }} ${{ steps.files.outputs.files_updated }}" >> $GITHUB_ENV
+      - name: Set outputs
+        id: set-modified
+        run: echo "modified=${{ env.modified }}" >> $GITHUB_OUTPUT
+      - name: Set versions
+        id: set-versions
+        working-directory: docker
+        run: |
+          echo "pth_version=$(python -c "import configparser; cfg=configparser.ConfigParser(); cfg.read('docker.cfg'); print(cfg.get('DEFAULT', 'build_docker_image_pytorch_version'))")" >> $GITHUB_OUTPUT
+          echo "hvd_version=$(python -c "import configparser; cfg=configparser.ConfigParser(); cfg.read('docker.cfg'); print(cfg.get('DEFAULT', 'build_docker_image_hvd_version'))")" >> $GITHUB_OUTPUT
+
+  build-hvd:
+    name: Build all Horovod flavoured PyTorch-Ignite images
+    needs: setup
+    if: github.event_name != 'pull_request' || (contains(needs.setup.outputs.modified, 'hvd/') || contains(needs.setup.outputs.modified, 'docker.cfg'))
+
+    # Cherry-picked from https://github.com/pytorch/test-infra/blob/main/.github/workflows/linux_job.yml
+    env:
+      REPOSITORY: ${{ github.repository }}
+      PR_NUMBER: ${{ github.event.pull_request.number }}
+    runs-on: linux.4xlarge
+    steps:
+      - name: Clean workspace
+        run: |
+          echo "::group::Cleanup debug output"
+          sudo rm -rfv "${GITHUB_WORKSPACE}"
+          mkdir -p "${GITHUB_WORKSPACE}"
+          echo "::endgroup::"
+
+      - name: Checkout repository (pytorch/test-infra)
+        uses: actions/checkout@v5
+        with:
+          # Support the use case where we need to checkout someone's fork
+          repository: pytorch/test-infra
+          path: test-infra
+
+      - name: Setup Linux
+        uses: ./test-infra/.github/actions/setup-linux
+
+      - name: Checkout repository (${{ github.repository }})
+        uses: actions/checkout@v5
+        with:
+          # Support the use case where we need to checkout someone's fork
+          repository: ${{ github.repository }}
+          ref: ${{ github.ref }}
+          path: ${{ github.repository }}
+          fetch-depth: 1
+
+      - name: Build docker images
+        working-directory: ${{ github.repository }}/docker
+        run: |
+
+          export PTH_VERSION=${{ needs.setup.outputs.pth_version }}
+          export HVD_VERSION=${{ needs.setup.outputs.hvd_version }}
+
+          bash build.sh hvd hvd-base
+          bash build.sh hvd hvd-vision
+          bash build.sh hvd hvd-nlp
+
+      - name: Publish docker images
+        if: github.event_name == 'workflow_dispatch' || github.event_name == 'release'
+        env:
+          DOCKER_TOKEN: ${{ secrets.DOCKER_TOKEN }}
+          DOCKER_USER: ${{ secrets.DOCKER_USER }}
+        working-directory: ${{ github.repository }}/docker
+        run: |
+          bash push_all.sh hvd-base
+          bash push_all.sh hvd-vision
+          bash push_all.sh hvd-nlp
+
+  build-hvd-apex:
+    name: Build all Horovod + Apex flavoured PyTorch-Ignite images
+    needs: setup
+    if: github.event_name != 'pull_request' || (contains(needs.setup.outputs.modified, 'hvd/') || contains(needs.setup.outputs.modified, 'docker.cfg'))
+
+    # Cherry-picked from https://github.com/pytorch/test-infra/blob/main/.github/workflows/linux_job.yml
+    env:
+      REPOSITORY: ${{ github.repository }}
+      PR_NUMBER: ${{ github.event.pull_request.number }}
+    runs-on: linux.12xlarge
+    steps:
+      - name: Clean workspace
+        run: |
+          echo "::group::Cleanup debug output"
+          sudo rm -rfv "${GITHUB_WORKSPACE}"
+          mkdir -p "${GITHUB_WORKSPACE}"
+          echo "::endgroup::"
+
+      - name: Checkout repository (pytorch/test-infra)
+        uses: actions/checkout@v5
+        with:
+          # Support the use case where we need to checkout someone's fork
+          repository: pytorch/test-infra
+          path: test-infra
+
+      - name: Setup Linux
+        uses: ./test-infra/.github/actions/setup-linux
+
+      - name: Checkout repository (${{ github.repository }})
+        uses: actions/checkout@v5
+        with:
+          # Support the use case where we need to checkout someone's fork
+          repository: ${{ github.repository }}
+          ref: ${{ github.ref }}
+          path: ${{ github.repository }}
+          fetch-depth: 1
+
+      - name: Build docker images
+        working-directory: ${{ github.repository }}/docker
+        run: |
+
+          export PTH_VERSION=${{ needs.setup.outputs.pth_version }}
+          export HVD_VERSION=${{ needs.setup.outputs.hvd_version }}
+
+          bash build.sh hvd hvd-apex
+          bash build.sh hvd hvd-apex-vision
+          bash build.sh hvd hvd-apex-nlp
+
+      - name: Publish docker images
+        if: github.event_name == 'workflow_dispatch' || github.event_name == 'release'
+        env:
+          DOCKER_TOKEN: ${{ secrets.DOCKER_TOKEN }}
+          DOCKER_USER: ${{ secrets.DOCKER_USER }}
+        working-directory: ${{ github.repository }}/docker
+        run: |
+          bash push_all.sh hvd-apex
+          bash push_all.sh hvd-apex-vision
+          bash push_all.sh hvd-apex-nlp
+
+  build-main:
+    name: Build all PyTorch-Ignite images
+    needs: setup
+    if: github.event_name != 'pull_request' || (contains(needs.setup.outputs.modified, 'main/') || contains(needs.setup.outputs.modified, 'docker.cfg'))
+
+    # Cherry-picked from https://github.com/pytorch/test-infra/blob/main/.github/workflows/linux_job.yml
+    env:
+      REPOSITORY: ${{ github.repository }}
+      PR_NUMBER: ${{ github.event.pull_request.number }}
+    runs-on: linux.4xlarge
+    steps:
+      - name: Clean workspace
+        run: |
+          echo "::group::Cleanup debug output"
+          sudo rm -rfv "${GITHUB_WORKSPACE}"
+          mkdir -p "${GITHUB_WORKSPACE}"
+          echo "::endgroup::"
+
+      - name: Checkout repository (pytorch/test-infra)
+        uses: actions/checkout@v5
+        with:
+          # Support the use case where we need to checkout someone's fork
+          repository: pytorch/test-infra
+          path: test-infra
+
+      - name: Setup Linux
+        uses: ./test-infra/.github/actions/setup-linux
+
+      - name: Checkout repository (${{ github.repository }})
+        uses: actions/checkout@v5
+        with:
+          # Support the use case where we need to checkout someone's fork
+          repository: ${{ github.repository }}
+          ref: ${{ github.ref }}
+          path: ${{ github.repository }}
+          fetch-depth: 1
+
+      - name: Build docker images
+        working-directory: ${{ github.repository }}/docker
+        run: |
+
+          export PTH_VERSION=${{ needs.setup.outputs.pth_version }}
+          export HVD_VERSION=${{ needs.setup.outputs.hvd_version }}
+
+          bash build.sh main base
+          bash build.sh main vision
+          bash build.sh main nlp
+
+      - name: Publish docker images
+        if: github.event_name == 'workflow_dispatch' || github.event_name == 'release'
+        env:
+          DOCKER_TOKEN: ${{ secrets.DOCKER_TOKEN }}
+          DOCKER_USER: ${{ secrets.DOCKER_USER }}
+        working-directory: ${{ github.repository }}/docker
+        run: |
+          bash push_all.sh base
+          bash push_all.sh vision
+          bash push_all.sh nlp
+
+  build-main-apex:
+    name: Build all PyTorch-Ignite images with Apex
+    needs: setup
+    if: github.event_name != 'pull_request' || (contains(needs.setup.outputs.modified, 'main/') || contains(needs.setup.outputs.modified, 'docker.cfg'))
+
+    # Cherry-picked from https://github.com/pytorch/test-infra/blob/main/.github/workflows/linux_job.yml
+    env:
+      REPOSITORY: ${{ github.repository }}
+      PR_NUMBER: ${{ github.event.pull_request.number }}
+    runs-on: linux.12xlarge
+    steps:
+      - name: Clean workspace
+        run: |
+          echo "::group::Cleanup debug output"
+          sudo rm -rfv "${GITHUB_WORKSPACE}"
+          mkdir -p "${GITHUB_WORKSPACE}"
+          echo "::endgroup::"
+
+      - name: Checkout repository (pytorch/test-infra)
+        uses: actions/checkout@v5
+        with:
+          # Support the use case where we need to checkout someone's fork
+          repository: pytorch/test-infra
+          path: test-infra
+
+      - name: Setup Linux
+        uses: ./test-infra/.github/actions/setup-linux
+
+      - name: Checkout repository (${{ github.repository }})
+        uses: actions/checkout@v5
+        with:
+          # Support the use case where we need to checkout someone's fork
+          repository: ${{ github.repository }}
+          ref: ${{ github.ref }}
+          path: ${{ github.repository }}
+          fetch-depth: 1
+
+      - name: Build docker images
+        working-directory: ${{ github.repository }}/docker
+        run: |
+
+          export PTH_VERSION=${{ needs.setup.outputs.pth_version }}
+          export HVD_VERSION=${{ needs.setup.outputs.hvd_version }}
+
+          bash build.sh main apex
+          bash build.sh main apex-vision
+          bash build.sh main apex-nlp
+
+      - name: Publish docker images
+        if: github.event_name == 'workflow_dispatch' || github.event_name == 'release'
+        env:
+          DOCKER_TOKEN: ${{ secrets.DOCKER_TOKEN }}
+          DOCKER_USER: ${{ secrets.DOCKER_USER }}
+        working-directory: ${{ github.repository }}/docker
+        run: |
+          bash push_all.sh apex
+          bash push_all.sh apex-vision
+          bash push_all.sh apex-nlp