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 OVERHAULS SERIALIZATION APIS WHILE GOOGLE CLOUD DROPS PYTHON 3.7 SUPPORT

By RepoJournal · Filed · About Google

JAX's PjRt layer now accepts SDY target versions during serialization, giving you fine-grained control over compatibility across versions.

The big shift on the JAX desk: SDY dialect target versioning is live [1]. Serialize() and SerializeUsingVersionedStablehlo() now accept an explicit SDY target version parameter, with GetDefaultSdyVersion() providing sensible defaults (following the same 12-week deprecation window as StableHlo). This matters because it lets you version-lock your compiled kernels independent of your JAX runtime. Meanwhile, JAX added the Faddeeva function (wofz) and extended erfcx to handle complex inputs [2], rounding out the scipy.special family. Shape polymorphism now has partial support for ragged_dot [3], though modes 1 and 2 still bail on dynamic group sizes (error messages now in place). The OneAPI GPU plugin skeleton landed [4], setting up Intel SYCL infrastructure with no kernel implementations yet, so don't expect GPU compute from this one. On the Google Cloud front, google-cloud-logging is cutting Python 3.7, 3.8, and 3.9 support [5], consolidating on 3.10+ and modernizing dependency chains. The gapic-generator shipped two releases [6][7] fixing coverage pragmas and URL routing in setup.py. Python-genai expanded computer_use field support for Vertex [8]. One test suite fix resolved a protobuf unpacking issue in BigTable admin tests [9].

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] [SDY] Update PjRt MLIR/HLO conversion APIs to accept a SDY target version during serialization google/jax
  2. [2] add wofz (Faddeeva function) and extend erfcx to complex inputs ↗ google/jax
  3. [3] [shape_poly] Add partial support for lax.ragged_dot. ↗ google/jax
  4. [4] [OneAPI GPU] Add OneAPI plugin extension infrastructure and glue code ↗ google/jax
  5. [5] feat(logging): drop support for Python 3.7, 3.8, and 3.9 runtimes (#17276) googleapis/google-cloud-python
  6. [6] gapic-generator: v1.34.1 ↗ googleapis/google-cloud-python
  7. [7] gapic-generator: v1.34.0 ↗ googleapis/google-cloud-python
  8. [8] feat: additional computer_use field support for vertex. ↗ googleapis/python-genai
  9. [9] tests: resolve failure related to unpacking dictionary for protobuf messages ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on May 28, 2026?
JAX's PjRt layer now accepts SDY target versions during serialization, giving you fine-grained control over compatibility across versions. In total, 37 commits, 34 pull requests, and 2 releases landed.
Who contributed to Google on May 28, 2026?
4 developers shipped this update, including TheSkyentist, copybara-service[bot], mini-goel, and parthea.
What were the notable Google updates?
[SDY] Update PjRt MLIR/HLO conversion APIs to accept a SDY target version during serialization, add wofz (Faddeeva function) and extend erfcx to complex inputs, and [shape_poly] Add partial support for lax.ragged_dot.

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?