140 wires and counting

$ follow Google

Keep up with Google in about 3 minutes: what actually shipped — the commits, pull requests, releases, and security advisories that matter.

or

fair warning: these emails are deeply technical. diffs, version numbers, CVEs, benchmark deltas. if that's not your idea of a good read, this isn't your newsletter.

Folds into your digest — weekly by default, monthly if you prefer. Unsubscribe in one click.

$ status

wire 2026-08-15
stories 62

© 2026 RepoJournal Home Showcase How it works Privacy

$ the-wire · showcase

JAX SHIPS NATIVE TPU CONVOLUTION SUPPORT, PYTHON 3.15 CONSTRAINT LIFTED

By RepoJournal · Filed · About Google · Composed from the cited sources · methodology

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.

The headline feature is Pallas convolution support [1], which brings pltpu.conv and lax.conv_general_dilated to TPU inference and training without falling back to slower decomposition paths. This lands Part 1 of the work: the TPU_ConvOp dialect definition, verification, Python bindings, and canonicalization into window slicing and matmul lowering. Dilations and negative padding follow in Part 2. Separately, Peter Hawkins lifted the Python 3.15 version constraint on libtpu [2] now that libtpu 0.0.46 ships with 3.15 wheels, clearing the path for JAX on the latest Python. JAX also modernized its type hints, replacing typing.Union and typing.Optional with PEP 604 syntax [3]. Over on the Genai side, python-genai added data retention controls to ToolParallelAiSearch alongside step_count for reinforcement tuning and BidiGenerateContentSetup [4], while AudioTranscriptionConfig now exposes VERBATIM and SMART modes [5]. On the cloud-python front, documentation is now published to https://cloud.google.com/python/docs/reference [6] under Client libraries help, and a packaging bug in google-cloud-audit-log that was installing unintended top-level docs was fixed [7].

Action items

References

  1. [1] 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
  2. [2] Regenerate requirements lock files. ↗ google/jax
  3. [3] [typing] upgrade uses of typing.Union and typing.Optional ↗ google/jax
  4. [4] feat: Add enable_data_retention to ToolParallelAiSearch, Add step_count to ReinforcementTuningHyperParameters, Add BidiGenerateContentSetup ↗ googleapis/python-genai
  5. [5] feat: Add `mode` enum (`VERBATIM`, `SMART`) to `AudioTranscriptionConfig` and `TranscriptionConfig`. ↗ googleapis/python-genai
  6. [6] feat(docs): publish Client libraries help to reference documentation ↗ googleapis/google-cloud-python
  7. [7] fix: `google-cloud-audit-log` wheel installs unintended top-level `docs` package (#17271) ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on August 15, 2026?
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. In total, 32 commits and 30 pull requests landed.
Who contributed to Google on August 15, 2026?
10 developers shipped this update, including Peter Hawkins, copybara-service, jax authors, Google Team Member, Sara Robinson, bshaffer, Олександр Федоров, and Mr-Documents, and 2 more.
What were the notable Google updates?
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, Regenerate requirements lock files, and [typing] upgrade uses of typing.Union and typing.Optional.