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

Topics: AI / ML Python Full archive →

The Wire · Showcase

JAX SHIPS MATRIX ALGEBRA TOOLKIT, PALLAS CACHE CUTS COMPILE TIME

By RepoJournal · Filed · About Google

JAX v0.10.1 lands four new linear algebra functions and performance optimizations that reshape how numerical code compiles, while the Genai client adds Gemini 3.5 Flash support.

JAX v0.10.1 [1] is out with a full linear algebra refresh. You now have `hadamard`, `circulant`, `dft`, and `leslie` matrix constructors in `jax.scipy.linalg`, plus TensorFlow-compatible AREA resizing in image operations. This isn't cosmetic. If you're building scientific computing stacks, you just deleted boilerplate. The real performance story is upstream: Pallas lowering now caches MLIR compilation [2], meaning repeated kernel patterns skip Python work and reuse inlined function definitions. For teams running heavy numerical workloads, compile-time wins stack up fast. Meanwhile, the python-genai client shipped v2.5.0 [3] with Gemini 3.5 Flash model support, plus new status fields for budget tracking [4]. Across google-cloud-python, the storage SDK now supports Object Contexts metadata [5], BigQuery dropped Python 3.7-3.9 support [6], and BigFrames added AEAD scalar functions [7]. That Python 3.10 floor matters if you're on older runtimes. A rendering fix for nested JSON structures in DataFrames [8] unblocks anyone hitting visualization crashes with complex data.

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] JAX v0.10.1 ↗ google/jax
  2. [2] Add a cache around Pallas MLIR lowering. google/jax
  3. [3] v2.5.0 ↗ googleapis/python-genai
  4. [4] feat: Add budget_exceeded status ↗ googleapis/python-genai
  5. [5] feat(storage): add object contexts in Python GCS SDK ↗ googleapis/google-cloud-python
  6. [6] feat(bigquery): drop Python 3.7-3.9 support and regenerate ↗ googleapis/google-cloud-python
  7. [7] feat: add `bigframes.bigquery.aead.*` scalar functions ↗ googleapis/google-cloud-python
  8. [8] fix: cast JSON and nested struct columns to string for anywidget rendering ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on May 21, 2026?
JAX v0.10.1 lands four new linear algebra functions and performance optimizations that reshape how numerical code compiles, while the Genai client adds Gemini 3.5 Flash support. In total, 34 commits, 35 pull requests, and 2 releases landed.
Who contributed to Google on May 21, 2026?
5 developers shipped this update, including copybara-service[bot], nidhiii-27, chalmerlowe, tswast, and shuoweil.
What were the notable Google updates?
JAX v0.10.1, Add a cache around Pallas MLIR lowering, and v2.5.0.

More from Google

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

Want every project, not just this one?