RepoJournal

$ cat google/week/2026-08-31.log

Google

Google

the week in review · Aug 31 – Sep 6, 2026

JAX free-threading push dominates the week

♥

JAX landed PyArray thread-safety, XLA pinning, Pallas tags, and removed deprecated Pallas APIs, all without a security advisory.

183 commits 167 PRs merged 16 releases 7 briefings covered
← previous week google wire →

What shipped: JAX dominated the period with a coordinated push on thread-safety and free-threading readiness. Highlights include a thread-safe PyArray and refactored array construction [1], a new jax::ft_mutex with thread-safety annotations [2], removal of deprecated Pallas APIs [3], and a lock-protected arg_exclude_mask fix [4]. JAX also pinned XLA to a new revision [5] and rejected malformed PyTreeDefProto instead of crashing [6]. python-genai v2.21.0 added streaming downloads [7], while google-cloud-python added Firestore gRPC message length limits [8] and Django 6.0 support for Spanner [9].

Why it matters: For JAX users, the new thread-safe PyArray and the removal of skip_checks [10] signal a hardening of the internals that will affect custom extensions or code relying on the old PyArray constructor. "Revert" of the apptopology addition [11] means that client library is not ready, so avoid depending on it.

Also: A Pallas update added an optional hashable tag for get_barrier_semaphore [12], and the python-genai SDK also added translation_config support [13] and audio/webm MIME support [14].

References

  1. [1] Add `PyArray::ifrt_array_ref()`, which returns `xla::ifrt::ArrayRef` instead of a raw pointer. ↗ google/jax
  2. [2] feat: Add support for the audio/webm MIME type. ↗ googleapis/python-genai
  3. [3] v2.21.0 ↗ googleapis/python-genai
  4. [4] Initialize arg_exclude_mask with the correct values if exclude_argnames is None ↗ google/jax
  5. [5] Update XLA dependency to use revision http://github.com/openxla/xla/commit/496bd4bd49db9ecbffd85da630b49c860b724604 ↗ google/jax
  6. [6] feat: Add translation_config SDK support for GEAP. ↗ googleapis/python-genai
  7. [7] Initialize arg_exclude_mask with the correct values if exclude_argnames is None ↗ google/jax
  8. [8] Add jax::ft_mutex and jax::ft_lock_guard with thread-safety annotations. ↗ google/jax
  9. [9] [pallas] Optional hashable tag in `pltpu.get_barrier_semaphore(tag=...)`. ↗ google/jax
  10. [10] feat: Add translation_config SDK support for GEAP. ↗ googleapis/python-genai
  11. [11] feat(django-google-spanner): support Django 6.0 ↗ googleapis/google-cloud-python
  12. [12] Reject malformed PyTreeDefProto instead of crashing ↗ google/jax
  13. [13] feat(firestore): configure gRPC message length limits for large documents ↗ googleapis/google-cloud-python
  14. [14] [JAX] Remove _check_and_rearrange and skip_checks from PyArray constructor. ↗ google/jax

$ ls google/week/ # the briefings behind this review

Mon Aug 31 JAX fixes arg_exclude_mask initialization and updates XLA dependency Tue Sep 1 python-genai 2.21.0 ships streaming downloads; JAX hardens deserialization and locking Wed Sep 2 JAX refactors array construction for free-threading safety Thu Sep 3 Bigtable reroutes CheckAndMutateRows and ReadModifyWriteRows Fri Sep 4 JAX removes deprecated Pallas APIs, adds barrier-skip for multi-device calls Sat Sep 5 JAX ships thread-safe PyArray, changes jax.jit inline semantics Sun Sep 6 JAX pins XLA to new revision
← previous week google wire →