RepoJournal
Google Google
114 wires and counting

$ follow Google

Keep up with Google in about 3 minutes: what actually shipped — the commits, pull requests, releases, and security advisories that matter.

or

fair warning: these emails are deeply technical. diffs, version numbers, CVEs, benchmark deltas. if that's not your idea of a good read, this isn't your newsletter.

Folds into your digest — weekly by default, monthly if you prefer. Unsubscribe in one click.

$ status

wire 2026-09-08
stories 31
RSS Archive Topics

© 2026 RepoJournal Home Showcase Explore How it works Privacy

$ the-wire · showcase

JAX removes dead Pallas flags and fixes removal errors

By RepoJournal · Filed · About Google

Sergei Lebedev Chris Jones
2 people shipped this
Sergei Lebedev @superbobry 3 cited Chris Jones @chr1sj0nes 2 cited

JAX cleaned up Pallas internals by removing two obsolete flags and a deprecated MemorySpace constant, while improving error messages for the APIs that remain.

Pallas deprecations got sharper. Accessing `MemorySpace.HOST` previously raised a generic `ValueError` because the deprecation was never registered in `jax/_src/deprecations.py`; the code now removes `pltpu-memory-space-host` entirely, since the API compatibility window has passed [1]. The `jax_pallas_use_mosaic_gpu` flag is also gone, as Mosaic GPU no longer supports `pl.pallas_call` [2].

MGpu tests and utilities were tidied in the same pass: `test_parity` drops a redundant condition around `tcgen05.fence`, which is a compiler-level directive, and `utils.py` loses a redundant check [3][4]. The change to removed Pallas APIs should now produce improved error messages [5].

$ ls google/reviews/ # the step-back reads

the week in review JAX free-threading push dominates the week the month in review JAX ships 0.11.1 and reworks custom_vjp, Cloud Python hardens Spanner
Topics: AI / ML Python Full archive →

Quick answers

What shipped in Google on September 8, 2026?
JAX cleaned up Pallas internals by removing two obsolete flags and a deprecated MemorySpace constant, while improving error messages for the APIs that remain. In total, 16 commits and 15 pull requests landed.
Who contributed to Google on September 8, 2026?
2 developers shipped this update, including Sergei Lebedev and Chris Jones.
What were the notable Google updates?
[pallas:mosaic] Removed `pltpu-memory-space-host`, [pallas:mgpu] Removed `jax_pallas_use_mosaic_gpu`, and [mgpu] Remove unnecessary condition in `test_parity`.
♥

$ log Google --recent

Mosaic GPU cleanup plus neg_p alignment in JAX 5 shipped JAX pins XLA to new revision 1 shipped JAX ships thread-safe PyArray, changes jax.jit inline semantics 68 shipped JAX removes deprecated Pallas APIs, adds barrier-skip for multi-device calls 83 shipped

$ log --all # elsewhere on the wire

@openclaw Selective evidence collection and truthful reviewer prompts land in ClawSweeper AI Agents Cline desktop commands unblocked from slow backend startup

$ ls showcases/

Rails Django Laravel Node.js Go Vercel Vue.js Kubernetes HashiCorp Arch Linux Linux Anthropic

$ init your-desk

RepoJournal can write one of these for any repo you care about. Sign in with GitHub, pick your repos and sources, and your desk gets a wire like this one every morning.