Who contributed to Google on July 9, 2026?

3 developers shipped this update, including mattjj, copybara-service[bot], and nbayati.

What were the notable Google updates?

[hijax] custom_vjp3 fixes: VmapOf sym-zero jvp, errors, pretty-printing, [hijax] custom_vjp3 fixes: VmapOf sym-zero jvp, errors, pretty-printing, and [pallas:mgpu] `plgpu.copy_gmem_to_smem` now supports `cp.async`.

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 FIXES CUSTOM_VJP3 AUTODIFF BUGS; PYTHON-GENAI BREAKS BACKWARD COMPATIBILITY

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

3 people shipped this

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

mattjj @mattjj 1 cited

nbayati @nbayati 1 cited

JAX shipped critical fixes to custom_vjp3 that were producing wrong-shaped outputs under vmap-of-jvp, while python-genai landed a breaking change removing three parameters and reshaping safety settings.

The JAX team patched custom_vjp3 to correctly handle symbolic-zero tangent avals in VmapOf.jvp [1], a bug that produced misshapen outputs when constants flowed through jvp-of-vmap combinations. The same PR also restored proper error messaging for jvp of symbolic_zeros=True custom_vjp, replacing bare NotImplementedError with the expected TypeError [2]. Separately, JAX's Pallas GPU support expanded with cp.async backing for copy_gmem_to_smem [3], controlled by a new impl parameter and supporting barriers for async waits. Over in python-genai, a major refactor [4] removes cached_content, presence_penalty, and frequency_penalty entirely while exposing safety_settings and labels without private SDK gates. This is a breaking change that will require client updates. The genai SDK also dropped the default api-revision header [5] to fix upstream issues. BigFrames 2.45.0 shipped with ai.classify, ai.score, and ai.if_ accessors for BigQuery DataFrames [6], plus local UDF execution support. Authentication landed new mTLS helpers [7] for custom connection pools to load default client certificates and resolve the GOOGLE_API_USE_MTLS_ENDPOINT environment variable.

Action items

→ Update python-genai clients to remove cached_content, presence_penalty, frequency_penalty usage before next release googleapis/python-genai [immediate]
→ Upgrade JAX if using vmap-of-jvp with custom_vjp3; verify symbolic-zero handling in your gradients google/jax [plan]
→ Review BigFrames 2.45.0 for new ai.classify and local UDF execution in your pipelines googleapis/google-cloud-python [monitor]
→ Test mTLS helpers in google.auth.transport.mtls if building custom connection pools googleapis/google-cloud-python [plan]

References

[1] [hijax] custom_vjp3 fixes: VmapOf sym-zero jvp, errors, pretty-printing google/jax
[2] [hijax] custom_vjp3 fixes: VmapOf sym-zero jvp, errors, pretty-printing ↗ google/jax
[3] [pallas:mgpu] `plgpu.copy_gmem_to_smem` now supports `cp.async` ↗ google/jax
[4] refactor(interactions)!: remove cached_content, presence_penalty, and frequency_penalty; expose safety_settings and labels ↗ googleapis/python-genai
[5] chore: Remove default api-revision header googleapis/python-genai
[6] bigframes: v2.45.0 ↗ googleapis/google-cloud-python
[7] feat(auth): Implement python mtls helpers ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on July 9, 2026?: JAX shipped critical fixes to custom_vjp3 that were producing wrong-shaped outputs under vmap-of-jvp, while python-genai landed a breaking change removing three parameters and reshaping safety settings. In total, 39 commits, 32 pull requests, and 1 releases landed.
Who contributed to Google on July 9, 2026?: 3 developers shipped this update, including mattjj, copybara-service[bot], and nbayati.
What were the notable Google updates?: [hijax] custom_vjp3 fixes: VmapOf sym-zero jvp, errors, pretty-printing, [hijax] custom_vjp3 fixes: VmapOf sym-zero jvp, errors, pretty-printing, and [pallas:mgpu] `plgpu.copy_gmem_to_smem` now supports `cp.async`.

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.