Who contributed to Google on July 2, 2026?

4 developers shipped this update, including copybara-service[bot], daniel-sanche, tswast, and hebaalazzeh.

What were the notable Google updates?

Refactor Pallas interpret mode to better share between GPU and TPU, [pallas:mosaic] `pltpu.emit_pipeline` now uses a dataclass wrapping the indices, and Add tiling info to LayoutConstraint custom-call.

Google

JAX, the GenAI SDK, and the Cloud libs - Google's open source layer

Pick a date

Topics: AI / ML Python Full archive →

The Wire · Showcase

JAX PALLAS REFACTOR UNLOCKS GPU-TPU UNIFIED MEMORY MODEL

By RepoJournal · Filed 06:03 UTC on July 2, 2026 · About Google

4 people shipped this

copybara-service[bot] @copybara-service[bot] 4 cited

daniel-sanche @daniel-sanche 1 cited

tswast @tswast 1 cited

hebaalazzeh @hebaalazzeh 1 cited

JAX's Pallas layer just got a major architecture shift that tears down the walls between GPU and TPU execution, opening the door to warp specialization and eventual Sparsecore support.

The refactor [1] makes Pallas interpret mode generic over memory allocation and compute units, which means GPU cluster support and GPU warp specialization (splitting and coalescing warpgroups) are now viable. TPU engineers get a clear path to Sparsecore support down the road. In parallel, the mosaic layer tightened up [2] by having `pltpu.emit_pipeline` use a dataclass wrapping indices instead of raw tuples, unlocking richer metadata like flat step counters for explicit index calls. GPU kernel interpreter work continues with [4] multi-dimensional barrier support now live, and [3] adds tiling metadata to LayoutConstraint custom-calls. On the cloud side, BigQuery SSLError handling [5] got a critical fix: invalid requests no longer trigger futile 5-minute retry loops, and errors now surface with clearer context. Google Cloud Python also patched [6] its read client user agent reporting and adjusted [7] the GAPIC generator to use absolute paths in lazy module loading instead of runtime evaluation.

Action items

→ Review Pallas interpret mode refactor [ref:1] if you maintain GPU/TPU kernel code - this is foundational for next-gen optimizations google/jax [plan]
→ Upgrade google-cloud-bigquery to pick up SSLError fix [ref:6] before next production deploy googleapis/google-cloud-python [immediate]
→ Monitor Gemini 3.1 Flash Lite and 3.5 Flash support [ref:10] rollout in BigQuery DataFrames googleapis/google-cloud-python [monitor]

References

[1] Refactor Pallas interpret mode to better share between GPU and TPU ↗ google/jax
[2] [pallas:mosaic] `pltpu.emit_pipeline` now uses a dataclass wrapping the indices ↗ google/jax
[3] Add tiling info to LayoutConstraint custom-call ↗ google/jax
[4] [Pallas][GPU kernel interpreter] Support barriers of multi-dimensional shape, i.e. where `num_barriers` is a tuple. ↗ google/jax
[5] fix(bigquery): avoid SSLError retry loop ↗ googleapis/google-cloud-python
[6] fix: include amended user agent in read client ↗ googleapis/google-cloud-python
[7] chore(generator): use absolute paths in `__lazy_modules__` ↗ googleapis/google-cloud-python
[8] feat: support gemini-3.x models (#17615) googleapis/google-cloud-python

Quick answers

What shipped in Google on July 2, 2026?: JAX's Pallas layer just got a major architecture shift that tears down the walls between GPU and TPU execution, opening the door to warp specialization and eventual Sparsecore support. In total, 17 commits and 15 pull requests landed.
Who contributed to Google on July 2, 2026?: 4 developers shipped this update, including copybara-service[bot], daniel-sanche, tswast, and hebaalazzeh.
What were the notable Google updates?: Refactor Pallas interpret mode to better share between GPU and TPU, [pallas:mosaic] `pltpu.emit_pipeline` now uses a dataclass wrapping the indices, and Add tiling info to LayoutConstraint custom-call.

JAX SYNCS XLA COMPILER STACK

JAX updated its XLA dependency to pull in the latest compiler improvements and optimizations.

ai 1 shipped 1-min read

Google 2 days ago

JAX OVERHAULS AUTODIFF INTERNALS WITH CUSTOM_VJP3, EXPANDS MESH API

JAX shipped a major redesign of custom differentiation rules that simplifies how you define vector-Jacobian products, while simultaneously opening up mesh operations to the public API.

ai 24 shipped 1-min read

Google 3 days ago

JAX FIXES REMAT3 CONSTANT HANDLING CRASH, GOOGLE CLOUD PYTHON CENTRALIZES REST TRANSCODING

Matthew Johnson patched a critical remat3 regression where closed-over constants were silently dropped, causing immediate crashes in production rematerialization pipelines [ref:1].

ai 50 shipped 1-min read

Google 4 days ago

JAX TIGHTENS XLA OPTIMIZATION CONTROLS, GOOGLE CLOUD PYTHON SETS UP FEATURE GATES

JAX makes XLA optimization levels human-readable while Google Cloud Python client libraries prepare a unified feature gating system for observability.

ai 54 shipped 1-min read

Elsewhere on the wire

AI Agents about 21 hours ago

CLINE SHIPS FIVE CRITICAL AGENT FIXES AFTER END-TO-END TESTING BLITZ

Saoudrizwan just landed a cascade of fixes that prevent the agent loop from running wild, stop file reads from resolving to the wrong directory, and plug the compaction pipeline.

ai-agents 16 shipped 2-min read

Local LLMs about 21 hours ago

LLAMA.CPP SHIPS MINIMAX-M3 SPARSE ATTENTION, VLLM UNLOCKS HETEROGENEOUS PREFILL-DECODE

The inference stack gets a new model architecture and a fundamental shift in how prefill and decode can run on different hardware configurations.

+10

llms 125 shipped 1-min read

@CachyOS about 21 hours ago

GLIBC 2.43 REBUILD SHIPS, INSTALLER JUMPS TO C++26

CachyOS pushed glibc 2.43 through the pipeline while the CLI installer upgraded its C++ standard to C++26, signaling a major refresh across core infrastructure.

infra 97 shipped 1-min read

Elixir & Phoenix about 21 hours ago

ECTO FIXES SUBQUERY PLANNING IN JOINS, ELIXIR PATCHES IEX PARSING

Ecto's query planner now handles subqueries correctly in interpolated join filters, while Elixir fixes a syntax error that broke 'not in' as a continuation operator in IEx.