Who contributed to Google on July 21, 2026?

3 developers shipped this update, including copybara-service, Allan Renucci, and hebaalazzeh.

What were the notable Google updates?

[hijax] add an einsum hijax primitive, Add gather reduced rule and scatter-add unreduced rule. The rule currently is very restrictive. It only supports `x.at[idx].get()` where `x` is `reduced` and `idx` is fully replicated, and [Mosaic GPU] Add `MemRefReshapeOp` to the dialect.

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 SHIPS EINSUM PRIMITIVE AS FOUNDATIONS FOR DISTRIBUTED COMPUTING OVERHAUL

By RepoJournal · Filed 06:59 UTC on July 21, 2026 · About Google

3 people shipped this

copybara-service @copybara-service 7 cited

hebaalazzeh @hebaalazzeh 2 cited

Allan Renucci @allanrenucci

JAX's hijax layer now supports einsum as a first-class primitive, setting up the infrastructure for major distributed execution improvements across the stack.

In brief

JAX added an einsum hijax primitive. that's the foundation for distributing tensor contractions across devices without rewriting your math.
JAX now has gather/scatter rules for distributed arrays. reduces boilerplate for slicing and accumulating across replicated tensors.
Mosaic GPU dialect gained MemRefReshapeOp. one more piece toward compiling GPU kernels from higher-level JAX ops.
GenAI SDK now supports mTLS in custom clients via google.auth. encrypted secrets stay encrypted everywhere they're used.

The einsum primitive [1] doesn't hit the public API yet, but it's the foundation for making jnp.einsum route through JAX's custom lowering system, unlocking better control over how tensor operations execute on distributed hardware. Alongside this, JAX added gather and scatter reduction rules [2] with initial support for fully replicated indices on reduced arrays, a deliberate constraint while the team hardens the distributed semantics. The Mosaic GPU dialect continues hardening with a new MemRefReshapeOp [3] replacing the current two-operation reshape pattern, simplifying GPU code generation. Separately, test infrastructure improvements [4] now route all JAX tests through a unified loader that enables parallel execution. On the Python SDK side, the GenAI library opened up ASR fields to public API [5] and began shipping an environments service [6], while adding mTLS support [7] for custom clients. The google-cloud-python ecosystem standardized universe domain handling [8] into a single canonical helper, and added Python 3.15 support [9] by removing upper bound constraints in sqlalchemy-bigquery.

Action items

→ Monitor JAX einsum primitive rollout for public API timeline google/jax [monitor]
→ Verify your Python 3.15 compatibility in google-cloud dependencies googleapis/google-cloud-python [plan]
→ Review new GenAI environments service for existing integrations googleapis/python-genai [monitor]

References

[1] [hijax] add an einsum hijax primitive ↗ google/jax
[2] Add gather reduced rule and scatter-add unreduced rule. The rule currently is very restrictive. It only supports `x.at[idx].get()` where `x` is `reduced` and `idx` is fully replicated. ↗ google/jax
[3] [Mosaic GPU] Add `MemRefReshapeOp` to the dialect. ↗ google/jax
[4] Fix JAX tests to use the JAX test loader. ↗ google/jax
[5] feat: make ASR fields public in v1main and Interactions ↗ googleapis/python-genai
[6] feat: Add environments service to GenAI SDK. ↗ googleapis/python-genai
[7] feat: Support mTLS in custom client using google auth mtls.get_default_ssl_context ↗ googleapis/python-genai
[8] feat(api-core): add get_universe_domain helper to universe.py ↗ googleapis/google-cloud-python
[9] fix: sqlalchemy bigquery python 3.15 compat (#17788) googleapis/google-cloud-python

Quick answers

What shipped in Google on July 21, 2026?: JAX's hijax layer now supports einsum as a first-class primitive, setting up the infrastructure for major distributed execution improvements across the stack. In total, 39 commits and 38 pull requests landed.
Who contributed to Google on July 21, 2026?: 3 developers shipped this update, including copybara-service, Allan Renucci, and hebaalazzeh.
What were the notable Google updates?: [hijax] add an einsum hijax primitive, Add gather reduced rule and scatter-add unreduced rule. The rule currently is very restrictive. It only supports `x.at[idx].get()` where `x` is `reduced` and `idx` is fully replicated, and [Mosaic GPU] Add `MemRefReshapeOp` to the dialect.

JAX TIGHTENS PPERMUTE VALIDATION TO CATCH SILENT CORRECTNESS BUGS

JAX is shipping a correctness fix that will break code relying on undefined behavior, and that's exactly the point.

ai 9 shipped 1-min read

Google 2 days ago

JAX BACKWARD PASS NOW LOGS DEBUG DATA. CLOUD AUTH TIGHTENS mTLS PRECEDENCE

JAX's autodiff pipeline shipped structured logging from transpose rules, letting you inspect what's happening inside the backward pass without tanking performance.

ai 15 shipped 1-min read

Google 3 days ago

JAX ACCELERATES MULTI-GPU COMPUTE WHILE GOOGLE AUTH PLUGS CONNECTION LEAK

JAX shipped three critical compiler improvements for distributed training while google-auth patches a file descriptor exhaustion bug that could starve production services.

ai 68 shipped 2-min read

Google 5 days ago

JAX ADDS FP6 SUPPORT END-TO-END, PYTHON-GENAI SHIPS AGENT CONFIG

JAX landed full float6 format support across CPU and GPU backends, while the AI SDK gains AntigravityAgentConfig and a trigger resource in v2.12.0.

ai 88 shipped 1-min read

Elsewhere on the wire

@cachyos about 9 hours ago

NORWAY MIRROR DOWN, KERNEL STACK SHIPS MAINTENANCE RELEASES

CachyOS infrastructure is reporting a downed Norway mirror while the kernel and toolchain stacks push routine version bumps across the board.

infra 55 shipped 1-min read

Elixir & Phoenix about 9 hours ago

ECTO GAINS FRAGMENT COLUMN NAMING, PHOENIX PLUGS GET SMARTER

Ecto's new with_columns/2 lets you assign names to SQL fragments, while Phoenix now handles module plugs with options in pipe_through declarations.

elixir 14 shipped 1-min read

Mobile platforms about 9 hours ago

KOTLIN RIPS OUT INTERPRETER GUTS, FLUTTER PATCHES RENDERING PIPELINE

Kotlin is systematically removing its IR interpreter in a multi-commit cleanup that signals a major shift in how the compiler handles intermediate representation, while Flutter ships critical Vulkan safety fixes and expands autofill support.

mobile 115 shipped 2-min read

@dotnet about 9 hours ago

MAUI 11 PREVIEW 6 SHIPS WITH 190 COMMITS, MACIOS FIXES GENERATOR FLAKINESS

.NET MAUI 11.0.0-preview.6 landed with significant platform improvements while dotnet/macios tackles a critical test parallelization bug that was silently dropping thread-safety calls from generated code.