Unify JAX and Pallas unit test workflows to overcome the 20-workflow limit in CI #1889

Job	Run time
metadata	1s
arm64 / test-distribution (extra-only-distribution.sh)	6s
arm64 / test-distribution (mirror-only-distribution.sh)	5s
amd64 / test-distribution (extra-only-distribution.sh)	8s
arm64 / test-distribution (upstream-only-distribution.sh)	9s
amd64 / test-distribution (mirror-only-distribution.sh)	9s
amd64 / test-distribution (upstream-only-distribution.sh)	6s
arm64 / build-base / bump-manifest	3s
amd64 / build-base / bump-manifest	7s
arm64 / build-base / build-base	1s
amd64 / build-base / build-base	1m 24s
arm64 / build-jax	1s
arm64 / build-maxtext	1s
arm64 / build-levanter	1s
arm64 / build-pallas	1s
arm64 / build-t5x	1s
arm64 / build-pax	1s
arm64 / test-jax	1s
arm64 / test-pallas	1s
arm64 / build-rosetta-t5x	1s
arm64 / test-upstream-t5x	1s
arm64 / test-te	1s
arm64 / build-rosetta-pax	1s
arm64 / test-upstream-pax	1s
arm64 / test-rosetta-t5x	1s
arm64 / test-rosetta-pax	1s
arm64 / collect-docker-tags	1s
amd64 / build-jax	1s
amd64 / build-levanter	1s
amd64 / build-t5x	1s
amd64 / build-maxtext	1s
amd64 / build-pallas	1s
amd64 / build-pax	1s
amd64 / test-jax	1s
amd64 / build-rosetta-t5x	1s
amd64 / test-upstream-t5x	1s
amd64 / test-pallas	1s
amd64 / test-te	1s
amd64 / build-rosetta-pax	1s
amd64 / test-upstream-pax	1s
amd64 / test-rosetta-t5x	1s
amd64 / test-rosetta-pax	1s
amd64 / collect-docker-tags	1s
make-publish-configs	1s
finalize	1s
publish	1s
	2m 54s

Provide feedback