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 REWRITES DMA SEMAPHORE LAYER FOR TPU; DJANGO-GOOGLE-SPANNER DROPS LEGACY SUPPORT

By RepoJournal · Filed · About Google

Pallas TPU is simplifying its DMA synchronization model while Google Cloud Python draws a line at Django 3.2 and Python 3.9.

The JAX team shipped a major refactor across Mosaic/Pallas TPU to make enqueue/wait DMA semaphores optional [1] [2] [3], reversing earlier commits and reducing operational complexity in distributed training kernels. This lands alongside cluster async store support for multi-GPU Pallas [4], expanding the framework's reach into larger distributed workloads. On the Google Cloud side, django-google-spanner v5.0.0 [5] [6] is now live, dropping support for Django 3.2 and 4.2 while mandating Python 3.10+—a clean break that aligns with Django 5.2 support. The python-genai library fixed docstring generation for Image.from_file() kwargs [7] [8], a small but important fix for API documentation consistency. Supporting infrastructure work continued on google-cloud-python with Bigtable integration fixes [9] and Regional Access Boundary additions to the auth SDK [10].

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] Mosaic/Pallas TPU DMA semaphore refactor (Part 1)
  2. [2] Mosaic/Pallas TPU DMA semaphore refactor (revert)
  3. [3] Mosaic/Pallas TPU DMA semaphore refactor (second revert)
  4. [4] Pallas multi-GPU cluster async store support
  5. [5] django-google-spanner v5.0.0 release
  6. [6] django-google-spanner v5.0.0 release PR
  7. [7] python-genai Image.from_file() docstring fix
  8. [8] python-genai Image.from_file() docstring fix (GitHub)
  9. [9] google-cloud-python Bigtable integration fixes
  10. [10] Auth SDK Regional Access Boundary support

Quick answers

What shipped in Google on May 1, 2026?
Pallas TPU is simplifying its DMA synchronization model while Google Cloud Python draws a line at Django 3.2 and Python 3.9. In total, 34 commits, 32 pull requests, and 3 releases landed.

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?