Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2023–2025 Google LLC
# Copyright 2023–2026 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand All @@ -12,7 +12,12 @@
# See the License for the specific language governing permissions and
# limitations under the License.

name: Add Label
# This workflow automatically adds a "pull ready" label to a PR
# when it has received at least 2 approving reviews and all checks have passed.
# It runs on pull request review events, workflow run completion events for the "MaxText Package Tests" workflow,
# and can also be triggered manually via workflow dispatch.

name: Add Pull Ready Label

on:
workflow_run:
Expand Down
72 changes: 0 additions & 72 deletions .github/workflows/build_and_upload_images.sh

This file was deleted.

4 changes: 2 additions & 2 deletions .github/workflows/build_package.yml
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2025 Google LLC
# Copyright 2023-2026 Google LLC

# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand All @@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.

# This file defines a module for building and uploading a maxtext pacakge
# This workflow builds and uploads MaxText package
# based on the pyproject.toml at the current github workspace.

name: Build and Upload MaxText Package
Expand Down
16 changes: 16 additions & 0 deletions .github/workflows/check_docs_build.yml
Original file line number Diff line number Diff line change
@@ -1,3 +1,19 @@
# Copyright 2023-2026 Google LLC

# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at

# https://www.apache.org/licenses/LICENSE-2.0

# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# This workflow checks if the documentation build is successful.

name: Check the Documentation Build

on:
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2025 Google LLC
# Copyright 2023-2026 Google LLC

# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand All @@ -12,7 +12,8 @@
# See the License for the specific language governing permissions and
# limitations under the License.

# This workflow will build maxtext python package and run tests.
# This workflow orchestrates the CI pipeline for MaxText,
# including building the package, running tests, and notifying about failures.

name: MaxText Package Tests

Expand Down Expand Up @@ -308,7 +309,7 @@ jobs:
name: Investigate failed build # investigates failure of scheduled run and comments on tracking issue
needs: [tpu-tests, gpu-tests, cpu-tests, maxtext_jupyter_notebooks, maxtext_tpu_pathways_unit_tests, maxtext_tpu_pathways_integration_tests, notify_failure]
if: ${{ always() && contains(needs.*.result, 'failure') && github.event_name == 'schedule' }}
uses: ./.github/workflows/gemini-investigate.yml
uses: ./.github/workflows/gemini_investigate.yml
permissions:
contents: 'read'
id-token: 'write'
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2023–2025 Google LLC
# Copyright 2023–2026 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand All @@ -12,7 +12,9 @@
# See the License for the specific language governing permissions and
# limitations under the License.

name: CodeQuality
# This workflow runs code quality checks using pre-commit on pull requests.

name: Code Quality

on:
pull_request:
Expand Down
Original file line number Diff line number Diff line change
@@ -1,3 +1,18 @@
# Copyright 2023–2026 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# This workflow checks for broken links in the documentation.

name: Documentation Link Checker

Expand Down
Original file line number Diff line number Diff line change
@@ -1,3 +1,21 @@
# Copyright 2023–2026 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# This workflow acts as a central dispatcher for handling Gemini-related commands
# issued via PR comments, review comments, or issue comments.
# Depending on the command, it routes the request to the appropriate workflow for processing.

name: 'Gemini Dispatch'

on:
Expand Down Expand Up @@ -130,7 +148,7 @@ jobs:
needs: 'dispatch'
if: |-
${{ needs.dispatch.outputs.command == 'review' }}
uses: './.github/workflows/gemini-review.yml'
uses: './.github/workflows/gemini_review.yml'
permissions:
contents: 'read'
id-token: 'write'
Expand All @@ -144,7 +162,7 @@ jobs:
needs: 'dispatch'
if: |-
${{ needs.dispatch.outputs.command == 'invoke' }}
uses: './.github/workflows/gemini-invoke.yml'
uses: './.github/workflows/gemini_invoke.yml'
permissions:
contents: 'read'
id-token: 'write'
Expand All @@ -158,7 +176,7 @@ jobs:
needs: 'dispatch'
if: |-
${{ needs.dispatch.outputs.command == 'investigate' }}
uses: './.github/workflows/gemini-investigate.yml'
uses: './.github/workflows/gemini_investigate.yml'
permissions:
contents: 'read'
id-token: 'write'
Expand Down
Original file line number Diff line number Diff line change
@@ -1,3 +1,22 @@
# Copyright 2023–2026 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# This workflow handles the "investigate" command for Gemini,
# which is used to analyze and investigate test failures by
# gathering relevant logs and context, and then running a Gemini
# agent to assist in diagnosing the issue.

name: 'Gemini Failure Investigator'

on:
Expand Down
Original file line number Diff line number Diff line change
@@ -1,3 +1,21 @@
# Copyright 2023–2026 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# This workflow handles "invoke" commands from the Gemini Dispatch workflow,
# allowing for flexible execution of custom logic based on PR comments,
# review comments, issue comments, or label events.

name: 'Gemini Invoke'

on:
Expand Down
Original file line number Diff line number Diff line change
@@ -1,3 +1,19 @@
# Copyright 2023–2026 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# This workflow handles Gemini review commands issued via PR comments, review comments, or issue comments.

name: 'Gemini Review'

on:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,9 +13,9 @@
# limitations under the License.

# This workflow builds and pushes MaxText images for both TPU and GPU devices.
# It runs automatically daily at 12am UTC, on Pull Requests, or manually via Workflow Dispatch.
# It runs automatically daily at 12am UTC, or manually via Workflow Dispatch.

name: Build Images
name: Build Nightly Docker Images

on:
schedule:
Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/pypi_release.yml
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ jobs:
build_and_test_maxtext_package:
name: Build and Test MaxText Package
needs: [release_approval]
uses: ./.github/workflows/build_and_test_maxtext.yml
uses: ./.github/workflows/ci_pipeline.yml
secrets: inherit

build_docs:
Expand All @@ -51,7 +51,7 @@ jobs:

check_links:
name: Check Links in Documentation
uses: ./.github/workflows/linkcheck.yml
uses: ./.github/workflows/docs_link_check.yml
secrets: inherit

publish_maxtext_package_to_pypi:
Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/release_pipeline.yml
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2026 Google LLC
# Copyright 2023-2026 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand Down Expand Up @@ -46,7 +46,7 @@ jobs:
if: |
always() &&
(needs.release_approval.result == 'success' || needs.release_approval.result == 'skipped')
uses: ./.github/workflows/build_and_test_maxtext.yml
uses: ./.github/workflows/ci_pipeline.yml
secrets: inherit

build_release_candidate_images:
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2023–2025 Google LLC
# Copyright 2023–2026 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand All @@ -12,6 +12,8 @@
# See the License for the specific language governing permissions and
# limitations under the License.

# This workflow checks if a PR description contains a checklist, and fails if not.

name: Require Checklist
on:
pull_request:
Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/run_jupyter_notebooks.yml
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2026 Google LLC
# Copyright 2023-2026 Google LLC

# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand All @@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.

# This file defines a module for running jupyter notebooks against the built maxtext package.
# This workflow runs jupyter notebooks against the built MaxText package.

name: Run Jupyter Notebooks

Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/run_pathways_tests.yml
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2025 Google LLC
# Copyright 2023-2026 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand All @@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.

# This file runs unit tests with Pathways backend.
# This workflow runs unit tests with Pathways backend.

name: Run Pathways Tests

Expand Down
Loading
Loading