RepoJournal
Google Google
115 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-09-09
stories 69
RSS Archive Topics

© 2026 RepoJournal Home Showcase Explore How it works Privacy

$ the-wire · showcase

JAX adds frozendict pytrees for Python 3.15, log2 primitives

By RepoJournal · Filed · About Google

copybara-service Sergei Lebedev kranipa Mark Daoust Anthonios Partheniou release-please[bot] Tomo Suzuki
7 people shipped this
copybara-service @copybara-service 2 cited Sergei Lebedev @superbobry 1 cited kranipa @kranipa 1 cited Mark Daoust @MarkDaoust Anthonios Partheniou @parthea release-please[bot] @release-please[bot] Tomo Suzuki @suztomo

JAX prepares for Python 3.15 by treating built-in frozendict as a pytree container, and adds log2 as a first-class primitive.

JAX now supports Python 3.15's built-in `frozendict` as a pytree container, matching `dict` behavior: keys sorted as auxiliary data, values flattened as children [1]. This means code using `frozendict` with JAX transformations like `jax.tree.map` or `jax.jit` will work without custom pytree registrations. It fixes issue #40517.

`jax.lax.log2` is now a first-class primitive alongside `exp2`, with `jax.numpy.log2` lowering directly through it [2]. This change adds `log2_p` and `exp2_p` to transformation rules and docs, enabling more direct manipulation and lowering.

For the OneAPI GPU backend, new solver kernels add LU decomposition, real symmetric and complex Hermitian eigen-decomposition, and QR-based singular value decomposition [3]. Jacobi and Polar SVD variants are marked unsupported.

A deprecated `pltpu.HOST` was removed from `[pallas:mosaic]` [4], continuing the deprecation cycle.

$ ls google/reviews/ # the step-back reads

the week in review JAX free-threading push dominates the week the month in review JAX ships 0.11.1 and reworks custom_vjp, Cloud Python hardens Spanner
Topics: AI / ML Python Full archive →

Quick answers

What shipped in Google on September 9, 2026?
JAX prepares for Python 3.15 by treating built-in frozendict as a pytree container, and adds log2 as a first-class primitive. In total, 35 commits, 33 pull requests, and 1 releases landed.
Who contributed to Google on September 9, 2026?
7 developers shipped this update, including copybara-service, Sergei Lebedev, kranipa, Mark Daoust, Anthonios Partheniou, release-please[bot], and Tomo Suzuki.
What were the notable Google updates?
Add frozendict support to pytree for Python 3.15 (PEP 814), Add `jax.lax.log2` and primitive `jax.lax.log2_p`, and [OneAPI GPU] Add OneAPI Solver getrf, syevd, heevd, and gesvd Kernels.
♥

$ log Google --recent

JAX removes dead Pallas flags and fixes removal errors 31 shipped Mosaic GPU cleanup plus neg_p alignment in JAX 5 shipped JAX pins XLA to new revision 1 shipped JAX ships thread-safe PyArray, changes jax.jit inline semantics 68 shipped

$ log --all # elsewhere on the wire

@vuejs Compiler-SFC fix restores await scope in nested blocks @vercel Next.js fixes dev startup race as Turborepo layers telemetry and matcher speedups

$ ls showcases/

Rails Django Laravel Node.js Go Vercel Vue.js Kubernetes HashiCorp Arch Linux Linux Anthropic

$ init your-desk

RepoJournal can write one of these for any repo you care about. Sign in with GitHub, pick your repos and sources, and your desk gets a wire like this one every morning.