diff --git a/.github/workflows/publish.yaml b/.github/workflows/publish.yaml
index d9999000df..a11a49c7cf 100644
--- a/.github/workflows/publish.yaml
+++ b/.github/workflows/publish.yaml
@@ -19,7 +19,7 @@ on:  # yamllint disable-line rule:truthy
   # postsubmit
   push:
     branches:
-    - main
+    - arm-build
     paths:
     - 'tensorflow_federated/version.py'
 
@@ -38,8 +38,8 @@ jobs:
     # * Repository is not a fork.
     # * Branch is `main` (for workflow_dispatch trigger).
     if: |
-      github.repository == 'google-parfait/tensorflow-federated'
-      && github.ref == 'refs/heads/main'
+      github.repository == 'JetBrains-Research/tensorflow-federated'
+      && github.ref == 'refs/heads/arm-build'
     runs-on: ubuntu-latest
     timeout-minutes: 5
     permissions:
@@ -98,7 +98,8 @@ jobs:
     name: Build Package
     needs: [publish-release]
     runs-on: ubuntu-20.04
-    timeout-minutes: 60
+    timeout-minutes: 360
+    environment: release
     steps:
 
     - name: Checkout repository
@@ -136,7 +137,7 @@ jobs:
         bazelisk run //tools/python_package:build_python_package \
             --build_tag_filters="-nokokoro,-nopresubmit,-requires-gpu-nvidia" \
             --google_credentials="${{ steps.auth.outputs.credentials_file_path }}" \
-            --remote_cache="https://storage.googleapis.com/tensorflow-federated-bazel-cache/${{ github.job }}" \
+            --remote_cache="https://storage.googleapis.com/${{ vars.BAZEL_CACHE_BUCKET }}/${{ github.job }}" \
             -- \
             --output_dir="${{ github.workspace }}/dist/"
 
@@ -194,6 +195,7 @@ jobs:
     timeout-minutes: 5
     permissions:
       id-token: write  # Required for trusted publishing.
+    environment: release
     steps:
 
     - name: Download Python package
@@ -204,3 +206,6 @@ jobs:
 
     - name: Publish Python package
       uses: pypa/gh-action-pypi-publish@v1.8.14
+      with:
+        password: ${{ secrets.PYTHON_REPOSITORY_TOKEN }}
+        repository-url: ${{ vars.PYTHON_REPOSITORY_URL }}
diff --git a/tensorflow_federated/version.py b/tensorflow_federated/version.py
index 33e6b8acdc..0c3e07cffb 100644
--- a/tensorflow_federated/version.py
+++ b/tensorflow_federated/version.py
@@ -13,4 +13,4 @@
 # limitations under the License.
 """TensorFlow Federated version."""
 
-__version__ = '0.88.0'
+__version__ = '0.88.0a1'