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.
Action items
- → Upgrade JAX to v0.10.1 if using image resizing or linear algebra - new functions ship stable google/jax [plan]
- → Pin python-genai to v2.5.0 and test Gemini 3.5 Flash model integration googleapis/python-genai [plan]
- → Plan Python 3.10 upgrade for google-cloud-python if on 3.7-3.9 - support ended googleapis/google-cloud-python [monitor]
- → Update google-cloud-storage SDK for Object Contexts feature parity googleapis/google-cloud-python [plan]
References
- [1] JAX v0.10.1 ↗ google/jax
- [2] Add a cache around Pallas MLIR lowering. google/jax
- [3] v2.5.0 ↗ googleapis/python-genai
- [4] feat: Add budget_exceeded status ↗ googleapis/python-genai
- [5] feat(storage): add object contexts in Python GCS SDK ↗ googleapis/google-cloud-python
- [6] feat(bigquery): drop Python 3.7-3.9 support and regenerate ↗ googleapis/google-cloud-python
- [7] feat: add `bigframes.bigquery.aead.*` scalar functions ↗ googleapis/google-cloud-python
- [8] fix: cast JSON and nested struct columns to string for anywidget rendering ↗ googleapis/google-cloud-python
FAQ
- What changed 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.
- What should Google teams do about it?
- Upgrade JAX to v0.10.1 if using image resizing or linear algebra - new functions ship stable • Pin python-genai to v2.5.0 and test Gemini 3.5 Flash model integration • Plan Python 3.10 upgrade for google-cloud-python if on 3.7-3.9 - support ended
- Which Google repositories shipped on May 21, 2026?
- google/jax, googleapis/python-genai, googleapis/google-cloud-python