Add protobuf to Orbax Export dependencies and install protobuf using python and ensure protoc installation in github workflow.

.github/workflows/build.yml

@@ -103,12 +103,21 @@ jobs:
       uses: actions/setup-python@0b93645e9fea7318ecaed2b359559ac225c90a2b  # v5.3.0
       with:
         python-version: ${{ matrix.python-version }}
+    - name: Install Protoc
+      uses: arduino/setup-protoc@v1
+      with:
+        version: '3.x'
     - name: Extract branch name
       shell: bash
       run: echo "branch=${GITHUB_HEAD_REF:-${GITHUB_REF#refs/heads/}}" >> $GITHUB_OUTPUT
       id: extract_branch
     - name: Install dependencies
       run: |
+        sudo apt-get update
+        sudo apt-get install -y protobuf-compiler
+
+        protoc -I=. --python_out=. $(find orbax/export/ -name "*.proto")
+
         pip install .
         pip install .[testing] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
         if [[ "${{ matrix.jax-version }}" == "newest" ]]; then

export/pyproject.toml

@@ -28,6 +28,7 @@ dependencies = [
     'jaxlib',
     'jaxtyping',
     'numpy',
+    'protobuf',
     "orbax-checkpoint >=0.9.0"
 ]