$ the-wire · showcase
JAX adds type-2 and type-3 Hermitian eigensolves, lexsort batching cuts unique compile time
By RepoJournal · Filed · About Google · Composed from the cited sources · methodology
![release-please[bot]](https://avatars.githubusercontent.com/in/40688?v=4)
9 people shipped this
JAX extended generalized Hermitian eigensolvers and fixed a compile-time blowup in jnp.unique, while the Google client libraries shipped a python-genai minor release and three api_core transport changes.
Support type-2 and type-3 generalized Hermitian eigenproblems google/jax
jax.scipy.linalg.eigh(a, b) used to reject type=2 and type=3; it now solves A B v = w v and B A v = w v via a Cholesky reduction, with the reduced matrix Lᴴ A L and eigenvectors transformed back by L⁻ᴴ for type 2 and L for type 3. Existing calls are unaffected; the new branches are the ones previously raising.
Sort lexsort keys in batches so compile time doesn't grow with the number of keys google/jax
For jnp.unique(x, axis=...), _unique_sorted_mask ordered slices with a single lexsort carrying one key per column, and the generated comparator grew with the key count. Keys are now sorted in batches, so compile time no longer scales with the number of keys; in the reported case of jnp.unique(jnp.zeros((2, 1000)), axis=0), nearly all of the roughly 15 seconds on CPU was compilation.
[BugFix] Replace phase-2 PartialReduce with Pad-Reshape-Reduce in SPMD. google/jax
When ApproxTopK is partitioned across the reduction dimension, the second reduction step previously emitted a secondary PartialReduce that could drop elements from trailing shards. It is now a Pad-Reshape-Reduce, which is where to look if you were seeing truncated results under SPMD partitioning.
v2.28.0 googleapis/python-genai
python-genai 2.28.0 adds lyria-3.5, gemini-omni-1.1-flash, and gemini-omni-flash-preview to the Interactions Model enum and supports continuation_token in Interactions, following the separate feature commit.
feat(auth): Support sync credentials in AsyncAuthorizedSession googleapis/google-cloud-python
AsyncAuthorizedSession now accepts standard credential types by wrapping them in an adapter, and the resumable upload path is being refactored to work with both aiohttp.ClientSession and AsyncAuthorizedSession instead of the former only.
