RepoJournal
Google

Google

JAX, the GenAI SDK, and the Cloud libs - Google's open source layer

Keep up with Google in about 3 minutes a day: what actually shipped — the commits, pull requests, releases, and security advisories that matter.

One email a day. Unsubscribe in one click.

Pick a date

Topics: AI / ML Python Full archive →

The Wire · Showcase

JAX SHIPS EMIT_PIPELINE PRIMITIVE, FIXES TPU CI FLAKINESS

By RepoJournal · Filed · About Google

JAX landed the emit_pipeline primitive base implementation [ref:1] while switching TPU CI to pinned versions to slash ephemeral test failures.

The emit_pipeline primitive [1] is a major foundation piece for the transform pipeline. JAX also added XLA Transform E2E testing [2] to catch integration issues before they hit production. On the reliability front, switching TPU CI jobs from nightly to pinned versions [3] eliminates the version skew debugging tax that was eating cycles on phantom failures. Pallas/GPU work continues: the kernel interpreter now properly handles collective_axes for run_scoped allocations [4], fixing a gap in distributed GPU kernel support. On the GenAI SDK side, automatic function calling in python-genai is now restricted to Chat.send_message [5], no longer available on direct Model.generate_content calls. This is a breaking change for anyone relying on the direct path. google-cloud-agentidentitycredentials shipped v0.1.0 [6], a new credential type for agent-based authentication workflows. BigFrames picked up Python bytecode translation [8] and better documentation on bqsql cell chaining [9], while Cloud Bigtable expanded sample_row_keys to support row_range filtering [7].

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] Add emit_pipeline primitive base implementation ↗ google/jax
  2. [2] Add XLA Transform E2E Test. ↗ google/jax
  3. [3] Switch TPU CI jobs to "pinned" instead of "nightly". ↗ google/jax
  4. [4] [Pallas][GPU kernel interpreter] Properly support `collective_axes` for allocations made by `run_scoped`. ↗ google/jax
  5. [5] test: for non-stream generate_content, automatic function calling is only available via Chat.send_message. Direct call to Model.generate_content does not support automatic function calling anymore. ↗ googleapis/python-genai
  6. [6] google-cloud-agentidentitycredentials: v0.1.0 ↗ googleapis/google-cloud-python
  7. [7] feat: support row_range in sample_row_keys method ↗ googleapis/google-cloud-python
  8. [8] chore(bigframes): Add simple Python bytecode translation ↗ googleapis/google-cloud-python
  9. [9] docs(bigframes): add a notebook explaining bqsql magics cell chaining ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on June 10, 2026?
JAX landed the emit_pipeline primitive base implementation [ref:1] while switching TPU CI to pinned versions to slash ephemeral test failures. In total, 42 commits, 39 pull requests, and 1 releases landed.
Who contributed to Google on June 10, 2026?
4 developers shipped this update, including copybara-service[bot], anvitTawar, TrevorBergeron, and tswast.
What were the notable Google updates?
Add emit_pipeline primitive base implementation, Add XLA Transform E2E Test, and Switch TPU CI jobs to "pinned" instead of "nightly".

More from Google

Daily updates, in your inbox

Follow Google

Keep up with Google in about 3 minutes a day: what actually shipped — the commits, pull requests, releases, and security advisories that matter.

or

One email a day. Unsubscribe in one click. Read a past issue →

Elsewhere on the wire

Want every project, not just this one?