copybara-service

Google

wire 2026-10-01

Vertex AI client timeout fix, Compute v1 API regenerated

• » fix: ensure Vertex AI interactions client uses default timeout=None instead of 5s. googleapis/python-genai

Google

wire 2026-09-30

Mosaic GPU drops downgradable transfers, gapic-generator reaches 1.41.0

• » [Mosaic GPU] Remove `DOWNGRADABLE` transfer mode and `_retry_on_failure`. google/jax
• » [Pallas/Mosaic GPU] Add an `optimized` kwarg to `atomic_*` store operations. google/jax

Google

wire 2026-09-28

JAX guards SparseCore tests behind libtpu 0.0.50 and pins a new XLA revision

• » Fix Windows frexp test normalization and guard SparseCore tests on older libtpu. google/jax
• » Add pvarys on operands that are not of floating dtype even when auto_pcast is False. This is because int/bool dtypes can't be differentiated so we won't see psum on backward pass (not true for floats) google/jax

Google

wire 2026-09-25

JAX splits Mosaic debug info to fix cache key mismatch; python-genai regenerates GAOS SDKs

• » [Mosaic TPU] Separate debug info from the Mosaic kernel `body` into a new `debug_locations` in `custom_call_config` with a new configuration flag `jax_mosaic_unzip_debug_locations`. google/jax
• » Dynamically derive TPU process count and bounds in multiprocess tests google/jax

Google

wire 2026-09-24

XLA:GPU to flip dot(f32,f32) default from TF32 to BF16

• » [XLA:GPU] In the nearest future we are going to flip the default precision of dot(f32,f32) from current TF32 precision to BF16. As a result of that the precision on GPU would match the precision on TP google/jax
• » chore: expose GenAI domain types (environments, triggers, webhooks, agents) in dedicated submodules googleapis/python-genai
• » Fix out-of-bounds slice check in NDIndexer when start is 0. google/jax

Google

wire 2026-09-23

JAX adds a polynomial primitive, Mosaic GPU grows partial reductions

• » Add `jax.lax.polynomial` / `polynomial_p` for polynomial evaluation. google/jax
• » [Mosaic GPU] Support partial reductions across warps and lanes google/jax

Google

wire 2026-09-22

JAX v0.12.0 cleanup lands, Mosaic GPU drops backend_config for symmetric memory

• » [Mosaic:GPU] Use frontend attributes to request collective memory for Mosaic GPU symmetric buffers. google/jax
• » Set the host value promise without holding Gil. google/jax

Google

wire 2026-09-21

JAX rewrites jnp.sinc for real inputs, tightens ULP bounds

• » Improve accuracy of real `jnp.sinc(x)` via additive range reduction. google/jax

Google

wire 2026-09-20

Voices API reaches the GAOS SDK, JAX moves its XLA pin

• » feat: add Voices API resource to GAOS SDK googleapis/python-genai

Google

wire 2026-09-19

JAX drops NumPy 2.1 and stale version gates

• » Measure real ULP errors in `numerics` tests (`elementary_test` and `special_test`). google/jax
• » Consolidate elementary and special function accuracy documentation. google/jax

Google

wire 2026-09-18

Spanner deadlock fixed, JAX 0.11.2 adds minmax and log2

• » [Pallas/Mosaic GPU] Add a `plgpu.store` API. google/jax
• » Add a batched version of jax.copy_to_host_async. google/jax

Google

wire 2026-09-17

JAX rewrites sinc near zero, python-genai ships v2.24.0

• » Fix jnp.sinc gradient and higher-order derivative accuracy near zero google/jax

Google

wire 2026-09-16

JAX accrues into reshaped refs, Bigtable reroutes ReadRows, Spanner Omni adds password auth

• » Support accumulation (`ref_addupdate`) on indexed `ReshapeTransform` views during state discharge, and explicitly reject unsupported view combinations. google/jax

Google

wire 2026-09-15

JAX lands batched host copies, Pallas MGPU deprecations

• » C++-side support for batched_copy_to_host_async. google/jax
• » [pltpu.emit_pipeline] Support async pipeline prefetch. google/jax

Google

wire 2026-09-13

JAX handles non-unimplemented layout errors, bumps XLA

• » Gracefully handle errors other than "unimplemented" when reading layouts. google/jax

Google

wire 2026-09-12

JAX pins down DCE jaxpr identity, fixes i0/i1 gradients at zero

• » Add `lax.one_minus_square` primitive to preserve accuracy of `1 - x^2` when evaluating near `+/-1` and differentiating near `0`. google/jax
• » Preserve memory space constraints on Ref buffers during state discharge. google/jax
• » feat: Add RetrievalCallStep and RetrievalResultStep to interactions schema and SDKs. googleapis/python-genai
• » fix: do not run functions once the automatic function calling budget is spent googleapis/python-genai

Google

wire 2026-09-11

Spanner drops an O(N^2) result loop, JAX adds ULP accuracy tests

• » Add ULP precision test suite for elementary math functions. google/jax

Google

wire 2026-09-10

google-auth 2.58.0 ships mTLS reconfiguration, JAX raises mpmath floor

• » Require mpmath>=1.4.1 and remove mpmath 1.3 workarounds. google/jax
• » Add LLO annotations to Pallas TPU google/jax
• » feat: support dynamic turn completion via interaction_status in live.py receive() googleapis/python-genai

Google

wire 2026-09-09

JAX adds frozendict pytrees for Python 3.15, log2 primitives

• » Add frozendict support to pytree for Python 3.15 (PEP 814). google/jax
• » Add `jax.lax.log2` and primitive `jax.lax.log2_p`. google/jax

Google

wire 2026-09-07

Mosaic GPU cleanup plus neg_p alignment in JAX

• » Aligned the lowering of `lax.neg_p` between Lane and WG semantics for floating types google/jax

Google

wire 2026-09-05

JAX ships thread-safe PyArray, changes jax.jit inline semantics

• » Make PyArray thread-safe. google/jax
• » Change inline=True in jax.jit to mean JAX_LATE instead of JAX_EARLY. google/jax

Google

wire 2026-09-04

JAX removes deprecated Pallas APIs, adds barrier-skip for multi-device calls

• » [Mosaic GPU] Add parameter to skip multi-device barriers passed through metadata. google/jax
• » Remove libtpu.so strings check from TPU CI scripts. google/jax
• » Guard PyArray_Storage::dynamic_shape with mu google/jax

Google

wire 2026-09-02

JAX refactors array construction for free-threading safety

• » [JAX] Remove _check_and_rearrange and skip_checks from PyArray constructor. google/jax
• » Add `PyArray::ifrt_array_ref()`, which returns `xla::ifrt::ArrayRef` instead of a raw pointer. google/jax

Google

wire 2026-09-01

python-genai 2.21.0 ships streaming downloads; JAX hardens deserialization and locking

• » feat: Add translation_config SDK support for GEAP. googleapis/python-genai
• » Add jax::ft_mutex and jax::ft_lock_guard with thread-safety annotations. google/jax
• » [pallas] Optional hashable tag in `pltpu.get_barrier_semaphore(tag=...)`. google/jax

Google

wire 2026-08-31

JAX fixes arg_exclude_mask initialization and updates XLA dependency

• » Initialize arg_exclude_mask with the correct values if exclude_argnames is None google/jax

Google

wire 2026-08-30

JAX ships tracing-time ordering, xla dependency bump

• » Use program_order_p even if exclude_argnames is None google/jax
• » Reverts 2610691a3fcfc16d5fd09e714f492f518978622c google/jax

Google

wire 2026-08-29

Google Cloud Python tools sync lint and format

• » feat: Add destination parameter to client.files.download to support streaming downloads to disk or file-like objects googleapis/python-genai
• » fix: expose ProcessingCallStep and ProcessingResultStep in Interactions SDK googleapis/python-genai

Google

wire 2026-08-28

JAX ships a GPU/TPU grab bag. Google Cloud goes shopping.

• » feat: add Video Understanding support to the Interactions API googleapis/python-genai
• » [Mosaic TPU] Add lowering rule for `lax.approx_max_k` in Pallas Mosaic. First reduce the reduction dimension to some number of bins, then do top_k. google/jax

Google

wire 2026-08-27

JAX Deletes shard_alike

• » Delete shard_alike from JAX. Use explicit mode as the replacement: https://docs.jax.dev/en/latest/201/sharding.html google/jax

Google

wire 2026-08-26

GENAI SHIPS 2.20.0 WHILE API-CORE PATCHES A YANKED RELEASE

• » [typing] Use PEP 695 typevars in jaxlib.pytree google/jax

Google

wire 2026-08-25

JAX restores fast multi-dimensional IRFFT on GPU

• » feat: add environment files support and scotty file download helper googleapis/python-genai
• » feat: Add AudioTranscriptionConfigMode googleapis/python-genai

Google

wire 2026-08-24

JAX SPEEDS UP SNAKE, DROPS ROCM WHEEL LOOKUP

• » Optimize `planar_snake` calculation. google/jax
• » Remove dynamic discovery of unknown rocm wheels. google/jax

Google

wire 2026-08-23

JAX Renames Mosaic GPU Parameters

• » [Mosaic:GPU] Rename multimem parameters mosaic attribute to symmetric memory parameters. google/jax

Google

wire 2026-08-22

GOOGLE WEB STACK AND SPANNER GET LAST-MINUTE ENGINEERING CATCH-UPS

• » chore: Disable GenerateImages for Gemini API googleapis/python-genai
• » Add machinery for LIT so we can have backend dependent layout rules (pallas TPU, pallas MGPU, JAX HLO, etc). google/jax
• » [fuser] Support static argnums and argnames. google/jax

Google

wire 2026-08-21

JAX hardens distributed mode with mTLS

• » Rename `VJPHiPrimitive` to `HiPrim` but keep `VJPHiPrimitive` as an alias google/jax
• » Migrate all usages of HiPrimitive to VJPHiPrimitive and delete HiPrimitive google/jax

Google

wire 2026-08-19

JAX CUTS PYTHON 3.11 AND BAZEL LEGACY IN ONE MORNING

• » [Pallas TPU] Roll back exposing accumulator references to be used with explicit MXU APIs. google/jax

Google

wire 2026-08-17

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

• » [Pallas TPU] Roll back exposing accumulator references to be used with explicit MXU APIs. google/jax

Google

wire 2026-08-16

JAX EXPANDS MPMD DEBUGGING, GOOGLE CLOUD PYTHON PLUGS BIGTABLE DATA LOSS

• » Support jax.debug.callback() in MPMD. google/jax
• » `core_map` now delegates to `mpmd_map` google/jax
• » Fix zizmor suppressions for ml_actions images to use the correct format. google/jax

Google

wire 2026-08-15

JAX SHIPS NATIVE TPU CONVOLUTION SUPPORT, PYTHON 3.15 CONSTRAINT LIFTED

• » Provide native convolution support (`pltpu.conv` and `lax.conv_general_dilated`) in JAX Pallas and the MLIR Mosaic TPU dialect for basic N-D spatial convolutions. google/jax
• » [typing] upgrade uses of typing.Union and typing.Optional google/jax
• » feat: Add enable_data_retention to ToolParallelAiSearch, Add step_count to ReinforcementTuningHyperParameters, Add BidiGenerateContentSetup googleapis/python-genai
• » feat: Add `mode` enum (`VERBATIM`, `SMART`) to `AudioTranscriptionConfig` and `TranscriptionConfig`. googleapis/python-genai

Google

wire 2026-08-13

JAX SWITCHES TO BZLMOD, SHIPS AOT COMPILATION FOR TRITON KERNELS

• » Implement AOT compilation for autotuned JAX-Triton kernels. google/jax
• » [triton] Added support for host-side TMA descriptors google/jax
• » [pallas:mgpu] Automatically infer `PROMISE_IN_BOUNDS` in `emit_pipeline` google/jax
• » fix: improve AFC(automatic function calling) in chat including fixing bugs in AFC for generate_content_stream. also log warnings in generate_content, generate_content_stream and their async variants t googleapis/python-genai

Google

wire 2026-08-12

JAX DROPS PYTHON 3.13T SUPPORT, CLEANS UP BUILD PIPELINE

• » Bump rules_python to v2.2.0. google/jax
• » Make dce_sink a VJPHiPrimitive so that it works with sharding google/jax
• » perf: build model validators on first use instead of at import googleapis/python-genai

Google

wire 2026-08-11

JAX WIRES INTEL GPU SUPPORT, FIXES MOSAIC ASSEMBLY BUGS

• » [NFC] Remove no-longer needed version guards google/jax
• » [pallas:mosaic] Reduced the number of usages of `pl.core_map` in tests google/jax

Google

wire 2026-08-10

JAX REVERTS BREAKING CHANGE, SYNCS XLA UPSTREAM

• » rollback of changelist 961233365. google/jax

Google

wire 2026-08-09

JAX FIXES ZERO-PROPAGATION BUG IN REVERSE-MODE AUTODIFF

• » Do not print the content of an array from `repr`/`str` if it's bigger than 1MiB google/jax

Google

wire 2026-08-08

JAX OVERHAULS REF SYSTEM AS GENAI SDK LOCKS IN JSON SCHEMA SUPPORT

• » feat: enable json schema in FunctionDeclaration parser googleapis/python-genai
• » feat: Add interaction_status to LiveServerContent googleapis/python-genai

Google

wire 2026-08-06

JAX PINS PYTHON 3.14T, GENAI GAINS ROBOTICS MODEL AND ERROR HANDLING

• » [Mosaic GPU] Define the `b6x16_p32` MLIR dtype. google/jax
• » [Pallas/Mosaic GPU][NFC] Deduplicate Pallas and Mosaic GPU compilation pipelines. google/jax
• » [Pallas/Mosaic GPU] Allow dumping module before and after layout inference. google/jax

Google

wire 2026-08-05

JAX ADDS NUMPY 2.6 TOP_K, INTEL GPU SUPPORT LANDS IN CI

• » [Mosaic TPU] Support lane dimension argmax/argmin with bf16 input. google/jax
• » Allow `reduced` operand to pass through `reduce_{max,min}`. Also add ur_rule for `div_p` since `reduce_max` jvp calls into div. google/jax

Google

wire 2026-08-04

JAX HARDENS GPU KERNELS WHILE CLOUD PYTHON PATCHES SUPPLY CHAIN RISK

• » [Mosaic GPU] Add support for transposed 8-bit stores (using ld/stmatrix) google/jax
• » [Mosaic GPU] Support print_layout for SMEM refs. google/jax
• » [pallas] Mirrored the `default_mesh_discharge_rule` memory space logic for `mpmd_map` google/jax
• » [pallas:mgpu] `copy_gmem_to_smem` no longer requires a barrier on pre-Hopper GPUs google/jax

Google

wire 2026-08-02

JAX AVERTS BREAKING CHANGE IN CUSTOM_GRADIENT API

• » Don't make `fun` in custom_gradient positional-only since that's a breaking change google/jax

Google

wire 2026-08-01

JAX REFACTORS CORE VJP SYSTEM WHILE CLOUD PYTHON FIXES SPANNER JSON BUGS

• » [pallas:mgpu] `copy_gmem_to_smem` now transparently uses TMA on Hopper+ and cp.async on older architectures google/jax

Google

wire 2026-07-31

JAX FIXES FP8 ATTENTION, LANDS TPU CUMULATIVE REDUCTIONS

• » [JAX] Lower cumulative reductions to chlo.ScanOp on TPU. google/jax

Google

wire 2026-07-30

JAX REBUILDS RUNTIME, GENAI SDK GAINS AUDIO TRANSCRIPTION

• » [Mosaic GPU] Move GPU module loading and function resolution out of runtime. google/jax
• » jaxpr.jaxpr->jaxpr google/jax
• » feat: Support GoogleMaps Tool grounding_types places and routing googleapis/python-genai
• » feat: Wire environment service into genai sdk googleapis/python-genai
• » feat: add environment resource googleapis/python-genai

Google

wire 2026-07-29

JAX STABILIZES STATE DISCHARGE, GOOGLE CLOUD PARALLELIZES SYSTEM TESTS 60 PERCENT FASTER

• » Refactor state discharge rules to use a context object and support stripping memory spaces. google/jax
• » Replace `is_cloud_tpu_older_than` with `is_libtpu_at_least` google/jax
• » feat: Add flat `language_codes` field to `AudioTranscriptionConfig`. googleapis/python-genai
• » test: update embedding, tuning tests googleapis/python-genai

Google

wire 2026-07-26

JAX OVERHAULS AUTODIFF INTERNALS WITH CUSTOM_VJP3, EXPANDS MESH API

• » Export `get_mesh`, `use_abstract_mesh` and `get_abstract_mesh` from the jax namespace google/jax

Google

wire 2026-07-25

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

• » [Mosaic GPU] Add layout inference rule for `MemRefReshapeOp`. google/jax
• » [Mosaic GPU] Refactor GPU Interpret Mode to use `GpuClockBundle`s google/jax

Google

wire 2026-07-24

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

• » Improve the way in which XLA optimization_level is set. google/jax
• » evaluate_sampler_bias: add support for discrete distributions google/jax
• » [pallas:sc] Added a bit more validation logic to abstract eval of various primitives google/jax

Google

wire 2026-07-23

GENAI AUDIO TRANSCRIPTION SHIPS, PROTO-PLUS FIXES RACE CONDITIONS

• » feat: [GenerateContentConfig] Add GenerationConfig.audio_transcription_config and Part.audio_transcription. googleapis/python-genai
• » [Mosaic GPU] Add shape checks for Pallas avals. google/jax

Google

wire 2026-07-22

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

• » Get rid of the plethora of async collective primitives we have. We can add more as needed. google/jax

Google

wire 2026-07-21

JAX SHIPS EINSUM PRIMITIVE AS FOUNDATIONS FOR DISTRIBUTED COMPUTING OVERHAUL

• » [hijax] add an einsum hijax primitive google/jax
• » 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
• » [Mosaic GPU] Add `MemRefReshapeOp` to the dialect. google/jax
• » Fix JAX tests to use the JAX test loader. google/jax
• » feat: make ASR fields public in v1main and Interactions googleapis/python-genai
• » feat: Add environments service to GenAI SDK. googleapis/python-genai
• » feat: Support mTLS in custom client using google auth mtls.get_default_ssl_context googleapis/python-genai

Google

wire 2026-07-20

JAX TIGHTENS PPERMUTE VALIDATION TO CATCH SILENT CORRECTNESS BUGS

• » Add `jax_raise_on_ppermute_sort_diff` enabled by default which errors out if axis_names passed to ppermute don't match the mesh axis_names order. google/jax

Google

wire 2026-07-18

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

• » [Pallas:MGPU] Add support for tcgen05.ld.red to warpgroup lowering google/jax
• » Add support for lax.split_p primitive in Pallas fuser google/jax
• » feat: add model selector googleapis/python-genai

Google

wire 2026-07-16

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

• » Add end-to-end support for FP6 formats (float6_e2m3fn and float6_e3m2fn) across CPU and GPU. google/jax
• » [Mosaic:GPU] Use XLA runtime API to run multi-device barrier kernel google/jax
• » [pallas] VJP rule for the einshape primitive google/jax

Google

wire 2026-07-15

JAX FIXES ROCM TRITON SUPPORT, CLOUD PYTHON TIGHTENS IMPORT PERFORMANCE

• » Expand copyright check to include shell scripts and CMake files. google/jax
• » chore: defer `mcp` SDK import to first MCP usage googleapis/python-genai

Google

wire 2026-07-14

JAX cuts rendezvous overhead in half with symmetric memory API, python-genai adds Triggers resource

• » [Mosaic:GPU] Use symmetric memory peer address API to get peer addresses in a single-process mode. google/jax
• » feat: Add Triggers resource to GoogleGenAI client. googleapis/python-genai
• » chore: Use trigger_id instead of id for run and list_execution method googleapis/python-genai

Google

wire 2026-07-12

JAX FIXES TYPING FAILURES AND SHARDING BUGS ACROSS CHECKPOINT AND EXTENDED DTYPE PIPELINES

• » Bring back SimpleNamespace export of checkpoint policies to fix typing failures google/jax
• » Convert out_shardings to physical shardings in presence of extended dtypes. The same happens for in_shardings. Fixes https://github.com/jax-ml/jax/issues/39089 google/jax
• » feat(interactions): add CodeMenderAgentConfig googleapis/python-genai
• » feat: Populate per-modality prompt token count in embedding responses for gemini-embedding-2 googleapis/python-genai

Google

wire 2026-07-11

JAX GPU KERNELS GET FASTER MEMORY ACCESS, BIGFRAMES UNLOCKS PANDAS-STYLE COLUMN INDEXING

• » [Mosaic GPU] Add basic support for ldmatrix/stmatrix google/jax