Who contributed to Google on May 24, 2026?

2 developers shipped this update, including copybara-service[bot] and gnecula.

What were the notable Google updates?

JAX: Add `mulhi` primitive which lowers to the recently-added `chlo.mulhi` op, [jax2tf] Clean up jax2tf examples and documentation, and Add a test for remat transpose when the primal inputs are reduced. Make sure the cotangents are unreduced.

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 ADDS HARDWARE-NATIVE INTEGER MULTIPLY, TIGHTENS jax2tf SCOPE

By RepoJournal · Filed 06:02 UTC on May 24, 2026 · About Google

2 people shipped this

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

gnecula @gnecula 1 cited

JAX shipped a new mulhi primitive that maps directly to the hardware instruction, while cleaning up jax2tf to focus on what it actually supports.

JAX landed the mulhi primitive [1], a new operation that lowers directly to the chlo.mulhi op, giving you access to native high-bit multiplication on supported hardware without the overhead of decomposition. This is a precision win for numerical work where you need both halves of the multiply result. In parallel, the jax2tf toolchain got a documentation and examples refresh [2] that strips out TensorFlow.js references and removes outdated serialization patterns, making it clearer what the bridge actually supports and reducing confusion for new users. The update doesn't break existing code but does expect you to drop any TF.js assumptions if you had them. On the testing front [3], JAX added hardening for remat transpose when primal inputs are reduced, ensuring cotangent handling stays correct through that optimization path.

Action items

→ Review jax2tf examples if you're using the bridge, especially TF.js paths jax-ml/jax [plan]
→ Test mulhi on your hardware target if you're doing integer-heavy numerics jax-ml/jax [monitor]

References

[1] JAX: Add `mulhi` primitive which lowers to the recently-added `chlo.mulhi` op. ↗ google/jax
[2] [jax2tf] Clean up jax2tf examples and documentation ↗ google/jax
[3] Add a test for remat transpose when the primal inputs are reduced. Make sure the cotangents are unreduced ↗ google/jax

Quick answers

What shipped in Google on May 24, 2026?: JAX shipped a new mulhi primitive that maps directly to the hardware instruction, while cleaning up jax2tf to focus on what it actually supports. In total, 4 commits and 3 pull requests landed.
Who contributed to Google on May 24, 2026?: 2 developers shipped this update, including copybara-service[bot] and gnecula.
What were the notable Google updates?: JAX: Add `mulhi` primitive which lowers to the recently-added `chlo.mulhi` op, [jax2tf] Clean up jax2tf examples and documentation, and Add a test for remat transpose when the primal inputs are reduced. Make sure the cotangents are unreduced.

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 1 day 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

Google 2 days ago

GENAI AUDIO TRANSCRIPTION SHIPS, PROTO-PLUS FIXES RACE CONDITIONS

Google's AI SDK gains native audio processing while the core Python cloud libraries ship critical stability fixes across the stack.

ai 60 shipped 1-min read

Google 3 days ago

JAX CUTS ASYNC COLLECTIVE PRIMITIVES, GENAI SDK ADDS LIVE API VOCABULARY

JAX is simplifying its async collective operations while the Python GenAI SDK ships custom vocabulary support for the Live API, marking a shift toward leaner abstractions and richer real-time capabilities.

ai 100 shipped 1-min read

Elsewhere on the wire

AI Agents about 9 hours ago

CLAUDE OPUS 5 LANDS ACROSS THE STACK

The newest Anthropic model is now live in langchain, Cline, and llama-index, with native support for extended reasoning and 1M context windows.

ai-agents 28 shipped 1-min read

Local LLMs about 9 hours ago

OLLAMA LANDS LAGUNA SUPPORT AND CRUSHES MEMORY LEAKS WHILE SGLANG HITS V0.5.16 WITH CONFIDENCE-DRIVEN SPECULATIVE DECODING

Ollama shipped three critical performance and reliability fixes for Metal residency and concurrent access patterns, while SGL-Lang released 0.5.16 with a new speculative algorithm hitting 383.7 tok/s on DeepSeek-V4.