Who contributed to Google on August 16, 2026?

4 developers shipped this update, including copybara-service, gulsumgudukbay, jax authors, and mutianf.

What were the notable Google updates?

Support jax.debug.callback() in MPMD, `core_map` now delegates to `mpmd_map`, and [ROCm] Fix ROCm version in plugin wheel metadata.

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 EXPANDS MPMD DEBUGGING, GOOGLE CLOUD PYTHON PLUGS BIGTABLE DATA LOSS

By RepoJournal · Filed 06:58 UTC on August 16, 2026 · About Google

4 people shipped this

copybara-service @copybara-service 3 cited

gulsumgudukbay @gulsumgudukbay 1 cited

mutianf @mutianf 1 cited

jax authors @Google-ML-Automation

JAX's callback system now works across multi-machine distributed setups, while Google Cloud Python fixes a critical bug that could silently drop database mutations.

JAX landed support for jax.debug.callback() in MPMD (multi-process multi-device) deployments [1], a capability that required plumbing host_callbacks through the IFRT compiler layer. This mirrors a larger architectural shift where core_map now delegates to mpmd_map [2], consolidating JAX's distributed execution paths. The framework also fixed ROCm version detection in hermetic wheel builds [3], which had been publishing incorrect metadata to PyPI since the build system migrated off rocm-jax. On the Cloud side, Google Cloud Python addressed a silent data loss vulnerability in Bigtable's batch operations [5], where flush errors and unacknowledged mutations could disappear without surfacing to application code. The fix adds explicit error reporting and validation to confirm all active requests complete before returning, preventing incomplete mutations from going undetected. Build tooling received attention too, with zizmor security scanning suppressions corrected to use proper formatting [4]. Across both codebases, 9 commits and 8 PRs landed in the last 24 hours.

Action items

→ Upgrade google-cloud-bigtable immediately if you run mutate_rows operations googleapis/google-cloud-python [immediate]
→ Test JAX callback debugging in your MPMD training setup google/jax [plan]

References

[1] Support jax.debug.callback() in MPMD. ↗ google/jax
[2] `core_map` now delegates to `mpmd_map` ↗ google/jax
[3] [ROCm] Fix ROCm version in plugin wheel metadata ↗ google/jax
[4] Fix zizmor suppressions for ml_actions images to use the correct format. ↗ google/jax
[5] fix(bigtable): report swallowed batch flush errors and unacknowledged entries ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on August 16, 2026?: JAX's callback system now works across multi-machine distributed setups, while Google Cloud Python fixes a critical bug that could silently drop database mutations. In total, 9 commits and 8 pull requests landed.
Who contributed to Google on August 16, 2026?: 4 developers shipped this update, including copybara-service, gulsumgudukbay, jax authors, and mutianf.
What were the notable Google updates?: Support jax.debug.callback() in MPMD, `core_map` now delegates to `mpmd_map`, and [ROCm] Fix ROCm version in plugin wheel metadata.

JAX ROLLS BACK TPU ACCUMULATOR API, CLOUD STORAGE FIXES STREAM CLEANUP

JAX reverted a Pallas TPU accumulator change after it broke the explicit MXU API surface, while Google Cloud Storage patched a critical gRPC stream cleanup bug that could leak resources on connection failures.

ai 5 shipped

Google Aug 15

JAX SHIPS NATIVE TPU CONVOLUTION SUPPORT, PYTHON 3.15 CONSTRAINT LIFTED

JAX Pallas now has native convolution operators for TPU, part one of a three-part overhaul that unlocks spatial N-D convolutions directly in the MLIR Mosaic dialect.

ai 62 shipped

Google Aug 14

JAX LOCKS IN PYTHON 3.15 SUPPORT AS GENAI SDK SHIPS VIDEO ENHANCEMENTS

JAX is ready for Python 3.15 and its free-threaded runtime, while Google's Genai SDK adds video resolution controls and BigQuery gets Apache Arrow query results.

ai 95 shipped

Google Aug 13

JAX SWITCHES TO BZLMOD, SHIPS AOT COMPILATION FOR TRITON KERNELS

JAX's build system migration to Bzlmod is live, paired with major performance improvements for autotuned GPU kernels that compile at instantiation rather than runtime.

ai 95 shipped

Elsewhere on the wire

AI Agents Aug 17

LANGCHAIN REFRESHES MODEL PROFILES AS OPENHANDS HARDENS LLM VALIDATION

LangChain's automated model profile system added DeepSeek-V4-Pro support while OpenHands deployed pre-flight LLM validation to catch misconfigured profiles before they ship.

ai-agents 8 shipped

Local LLMs Aug 17

LLAMA.CPP SHIPS MODEL OPTIMIZATIONS WHILE VLLM AND SGLANG CHASE MEMORY AND SCHEDULING GAINS

Across 87 commits and 5 releases, the local LLM stack is grinding through the unglamorous work that makes inference faster: killing dead code, fixing kernel launches, and removing synchronization bottlenecks.

llms 178 shipped

@CachyOS Aug 17

CACHYOS PACKAGES ROLL FORWARD, NANJING MIRROR GOES DARK

Python 3.11.16 and asusctl 6.4.0 ship across the stack as the China Nanjing mirror drops offline, signaling infrastructure pressure.

infra 64 shipped

Elixir & Phoenix Aug 17

ELIXIR CORE SHIPS CALENDAR SPEED SWEEP, PHOENIX LIVE VIEW FIXES UI STALLS

Three precision optimizations land in Elixir core with 1.75x speedups on date and time comparisons, while Phoenix Live View plugs a bug that was freezing UI for 200ms on zero-duration transitions.