RepoJournal
Google

Google

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

Pick a date

  1. Sun 3 may
  2. Mon 4 may
  3. Tue 5 may
  4. Wed 6 may
  5. Thu 7 may
  6. Fri 8 may
  7. Sat 9 may
  8. Sun 10 may
  9. Mon 11 may
  10. Tue 12 may
  11. Wed 13 may
  12. Fri 15 may
  13. Sat 16 may
  14. Sun 17 may
  15. Mon 18 may
  16. Tue 19 may
  17. Wed 20 may
  18. Thu 21 may
  19. Sat 23 may
  20. Sun 24 may
  21. Mon 25 may
  22. Tue 26 may
  23. Thu 28 may
  24. Fri 29 may
  25. Sat 30 may
  26. Sun 31 may
  27. Mon 1 jun
  28. Mon 8 jun
  29. Tue 9 jun
  30. Today 10 jun

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].

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

FAQ

What changed in Google on June 10, 2026?
JAX landed the emit_pipeline primitive base implementation while switching TPU CI to pinned versions to slash ephemeral test failures.
What should Google teams do about it?
Update python-genai code relying on Model.generate_content function calling - switch to Chat.send_message • Review emit_pipeline integration requirements for transform-heavy workloads • Integrate google-cloud-agentidentitycredentials if using agent-based auth patterns
Which Google repositories shipped on June 10, 2026?
google/jax, googleapis/python-genai, googleapis/google-cloud-python

Related across the cluster

For your repos

The showcase is a teaser.
Your wire is the product.

Same engine. Different stack. Below: what changes when the wire is yours.

Showcase wire

  • •14 famous open source orgs
  • •One wire per day
  • •Public, generic
  • •Read on the web, when you remember

Your wire

  • →Up to 1,500 of your repos - orgs, deps, vendors
  • →Morning and evening briefs
  • →Action items routed to your team
  • →Slack delivery, email, breaking-news CVE alerts

Want a hands-on demo first? Ask a current user for an invite link.