RepoJournal
Google

Google

JAX, the GenAI SDK, and the Cloud libs - Google's open source layer

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

One email a day. Unsubscribe in one click.

Pick a date

  1. Sat 18 jul
  2. Sun 19 jul
  3. Mon 20 jul
  4. Tue 21 jul
  5. Wed 22 jul
  6. Thu 23 jul
  7. Fri 24 jul
  8. Sat 25 jul
  9. Sun 26 jul
  10. Mon 27 jul
  11. Tue 28 jul
  12. Wed 29 jul
  13. Thu 30 jul
  14. Fri 31 jul
  15. Sat 1 aug
  16. Sun 2 aug
  17. Mon 3 aug
  18. Tue 4 aug
  19. Wed 5 aug
  20. Thu 6 aug
  21. Fri 7 aug
  22. Sat 8 aug
  23. Sun 9 aug
  24. Mon 10 aug
  25. Tue 11 aug
  26. Wed 12 aug
  27. Thu 13 aug
  28. Fri 14 aug
  29. Sat 15 aug
  30. Today 16 aug
Topics: AI / ML Python Full archive →

The Wire · Showcase

JAX SHIPS NATIVE TPU CONVOLUTION SUPPORT, PYTHON 3.15 CONSTRAINT LIFTED

By RepoJournal · Filed · About Google

copybara-service Peter Hawkins bshaffer Олександр Федоров jax authors Google Team Member Sara Robinson
10 people shipped this
copybara-service @copybara-service 4 cited Peter Hawkins @hawkinsp 1 cited bshaffer @bshaffer 1 cited Олександр Федоров @namefreezers 1 cited jax authors @Google-ML-Automation Google Team Member @google-genai-bot Sara Robinson @sararob Mr-Documents @Mr-Documents lyushher @lyushher ohmayr @ohmayr

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].

One email a day. Unsubscribe in one click.

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.

More from Google

Google Google

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

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.

copybara-service gulsumgudukbay jax authors mutianf
ai 17 shipped
Google Google

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.

hawkinsp jakevdp copybara-service mattjj +8
ai 95 shipped
Google Google

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.

hawkinsp copybara-service Google Team Member Annie Luc +3
ai 95 shipped
Google Google

JAX DROPS PYTHON 3.13T SUPPORT, CLEANS UP BUILD PIPELINE

JAX consolidated its dependency lock files after dropping Python 3.13 free-threading support, signaling a shift in the framework's build strategy as it pushes through infrastructure modernization.

Peter Hawkins copybara-service Benjamin Chetioui Amy Wu +4
ai 91 shipped

Daily updates, in your inbox

Follow Google

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

or

One email a day. Unsubscribe in one click. Read a past issue →

Elsewhere on the wire

AI Agents AI Agents

OPENHANDS PINS CLIENT DEPENDENCY, ADDS STRUCTURED ERROR HANDLING

OpenHands locked its TypeScript client to exact version 1.37.0 and merged canvas error outcome improvements in a single commit cycle.

neubig
ai-agents 2 shipped
Local LLMs Local LLMs

LLAMA.CPP OVERHAULS INFERENCE FLAGS, VLLM PATCHES DEEPSEEK SPARSE MLA

llama.cpp deprecates mmap flags in favor of unified load-mode, while vllm fixes critical issues blocking DeepSeek-V4-Flash across all decode paths.

github-actions[bot] dhiltgen Fathi Boudra Eric Zhang +11
llms 136 shipped
CachyOS @CachyOS

DASHBOARD SEO OVERHAUL SHIPS AS PACKAGE TRAIN ROLLS FORWARD

CachyOS public-dashboard refactored routes with canonical metadata and improved crawler visibility [ref:18] while 61 commits across five repos kept the AUR pipeline humming.

github-actions witchlliee Jinra azdanov +1
infra 72 shipped
Elixir & Phoenix Elixir & Phoenix

ELIXIR CORE CUTS ALLOCATIONS ACROSS STRING AND AST OPERATIONS

Thomas Cioppettini landed four performance wins overnight that eliminate unnecessary allocations in List.myers_difference, String case conversion, and Macro tree walkers.

Thomas Cioppettini LostKobrakai
elixir 17 shipped
Want every project, not just this one?