Kanglan Tang

Internal config change

Second test

Test internal config change

Test internal config change

ImportError: cannot import name 'oex_orchestration_pb2' from 'orbax.export.protos'

3

When running the latest `orbax-export`, the `from orbax.export.protos import oex_orchestration_pb2` line in utils.py is failing with an `ImportError: cannot import name 'oex_orchestration_pb2' from 'orbax.export.protos'`. Here is a test log: https://github.com/jax-ml/jax-ai-stack/actions/runs/17402977130/job/49400221219.

Generate requirement files and pyproject.toml for an AXLearn GPU workflow using seed-env

1

Steps: 1. Copy the necessary GPU container dependencies from AXLearn's pyproject.toml into a requirements.txt file, excluding seqio due to its outdated protobuf upper bound. 2. pip install seed-env 3. run...