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 HARDENS GPU KERNEL COMPILATION, CLOUD PYTHON SECURES mTLS PATH

By RepoJournal · Filed · About Google

JAX merged five critical infrastructure upgrades across Mosaic GPU and TPU while google-cloud-python closed a mTLS security gap that could silently fall back to insecure connections.

The JAX team shipped major plumbing work on the GPU side: FragmentedArray concatenation support [1] and PDL kernel launch integration [2] unlock more efficient Mosaic GPU compilation, while the TPU desk expanded dynamic rotate operations [5] to handle packed types with zero stride. These aren't user-facing features yet but they're the foundation for faster compiled kernels. Meanwhile, JAX is systematically removing internal APIs from jax.core [4], moving `abstract_token` to the public FFI layer ahead of the v0.11.0 cleanup. The error handling overhaul converting gpuError_t to absl::Status [6] tightens consistency across the stack. On the cloud side, google-cloud-python fixed a dangerous mTLS bug [3] where transport and token discovery could diverge, causing bound tokens to mismatch or fall back silently to unencrypted TLS. The version scanner got two quality-of-life upgrades: globbing support in ignore files [7] and generalized regex rules for dependency scanning [8]. BigFrames picked up AI classifier and scoring functions [9], and agent registry v1 is now available [10].

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] [Mosaic GPU] Add support for concatenating `FragmentedArray`s. ↗ google/jax
  2. [2] Add PDL kernel launch integration to pallas mosaic gpu ↗ google/jax
  3. [3] fix(auth): align mTLS discovery and enforce fail-fast transport configuration. ↗ googleapis/google-cloud-python
  4. [4] Expose `jax.ffi.abstract_token` google/jax
  5. [5] [Mosaic TPU] Support dynamic rotate on the second minor dim with zero stride for packed types, we require the dim size to be divisible by `packing x sublane_count`. ↗ google/jax
  6. [6] Convert all usages of gpuError_t to use absl::Status consistently (or convert). ↗ google/jax
  7. [7] feat(version-scanner): support globbing and subpath patterns in ignore file ↗ googleapis/google-cloud-python
  8. [8] feat(version-scanner): generalize and enhance regex rules for dependency scanning ↗ googleapis/google-cloud-python
  9. [9] feat(bigframes): add ai.classify, ai.score, ai.if_ to the df bq accessor ↗ googleapis/google-cloud-python
  10. [10] feat(google/cloud/agentregistry/v1): add google-cloud-agentregistry ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on June 27, 2026?
JAX merged five critical infrastructure upgrades across Mosaic GPU and TPU while google-cloud-python closed a mTLS security gap that could silently fall back to insecure connections. In total, 36 commits and 34 pull requests landed.
Who contributed to Google on June 27, 2026?
5 developers shipped this update, including copybara-service[bot], vverman, chalmerlowe, sycai, and JoeWang1127.
What were the notable Google updates?
[Mosaic GPU] Add support for concatenating `FragmentedArray`s, Add PDL kernel launch integration to pallas mosaic gpu, and fix(auth): align mTLS discovery and enforce fail-fast transport configuration.

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?

We use privacy-friendly analytics (Google Analytics, IP-anonymized) to see which pages help readers. No ads, and we never sell your data. See our Privacy Policy.