Who contributed to Google on August 25, 2026?

7 developers shipped this update, including hbahk, hawkinsp, mattjj, copybara-service, Vladimir Belitskiy, Mark Daoust, and release-please[bot].

What were the notable Google updates?

Lower multi-dimensional IRFFT on GPU to a single C2R transform again, [custom_vjp3] fix tracer leak in custom_vjp3 remat rule, and Regenerate lock files, notably for SciPy 1.18.1.

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 restores fast multi-dimensional IRFFT on GPU

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

7 people shipped this

release-please[bot] @release-please[bot] 5 cited

copybara-service @copybara-service 2 cited

hbahk @hbahk 1 cited

hawkinsp @hawkinsp 1 cited

mattjj @mattjj 1 cited

Vladimir Belitskiy @belitskiy

Mark Daoust @MarkDaoust

A JAX patch cuts multi-dimensional IRFFT time on GPU by roughly 1.7x, restoring the single C2R transform after years of slower decomposition.

JAX's GPU users get a performance win: a PR lowers multi-dimensional IRFFT back to a single C2R transform, recovering the speed lost since 0.10.0 while preserving NumPy semantics [1]. The fix closes issue #40143 and is a must-pick-up for anyone doing FFT-heavy GPU workloads. Elsewhere in JAX, a tracer leak in the custom_vjp3 remat rule was fixed, preventing the cryptic "No constant handler for DynamicJaxprTracer" error when enabling the new flags [2]. Lock files were regenerated for SciPy 1.18.1, which should unblock Python 3.15 CI [3]. Over in google-cloud-python, sqlalchemy-spanner hit v1.20.0, adding native UUID data type support and exporting the MAX_SIZE constant [4]. pandas-gbq v0.35.2 rejects backticks in table IDs, a small but smart hardening [5]. The python-genai client added environment file support and a Scotty file download helper [6], plus a new AudioTranscriptionConfigMode [7]. Release notes for the patch trio describe only documentation pointer changes, so skip those unless you need the pointer cleanup [8] [9] [10]. The JAX IRFFT restoration is the one to grab first.

Action items

→ Update JAX to include PR #40144 for the IRFFT speedup before your next FFT-heavy training run google/jax [immediate]
→ If you use custom_vjp3 with remat, pull in PR #40178 to avoid tracer leak crashes google/jax [immediate]
→ Upgrade sqlalchemy-spanner to v1.20.0 to get native UUID support googleapis/google-cloud-python [plan]
→ Adopt pandas-gbq v0.35.2 to reject backtick-injected table IDs googleapis/google-cloud-python [plan]

References

[1] Lower multi-dimensional IRFFT on GPU to a single C2R transform again ↗ google/jax
[2] [custom_vjp3] fix tracer leak in custom_vjp3 remat rule ↗ google/jax
[3] Regenerate lock files, notably for SciPy 1.18.1. ↗ google/jax
[4] sqlalchemy-spanner: v1.20.0 ↗ googleapis/google-cloud-python
[5] pandas-gbq: v0.35.2 ↗ googleapis/google-cloud-python
[6] feat: add environment files support and scotty file download helper ↗ googleapis/python-genai
[7] feat: Add AudioTranscriptionConfigMode ↗ googleapis/python-genai
[8] proto-plus: v1.28.4 ↗ googleapis/google-cloud-python
[9] googleapis-common-protos: v1.75.2 ↗ googleapis/google-cloud-python
[10] google-resumable-media: v2.10.2 ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on August 25, 2026?: A JAX patch cuts multi-dimensional IRFFT time on GPU by roughly 1.7x, restoring the single C2R transform after years of slower decomposition. In total, 42 commits, 35 pull requests, and 10 releases landed.
Who contributed to Google on August 25, 2026?: 7 developers shipped this update, including hbahk, hawkinsp, mattjj, copybara-service, Vladimir Belitskiy, Mark Daoust, and release-please[bot].
What were the notable Google updates?: Lower multi-dimensional IRFFT on GPU to a single C2R transform again, [custom_vjp3] fix tracer leak in custom_vjp3 remat rule, and Regenerate lock files, notably for SciPy 1.18.1.

JAX SPEEDS UP SNAKE, DROPS ROCM WHEEL LOOKUP

JAX's planar_snake is now up to 10% faster for power-of-two tiles, and the team is killing a painfully slow ROCm wheel discovery path.

ai 7 shipped

Google Aug 23

JAX Renames Mosaic GPU Parameters

Google's JAX team is renaming a core Mosaic GPU attribute, and it will break your cross-process parameter code.

ai 3 shipped

Google Aug 22

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

If you thought Google's Python ecosystem was quiet overnight, you missed a flurry of fixes that touch everything from Bigtable filters to Spanner's DBAPI driver.