$ the-wire · showcase
JAX ships a GPU/TPU grab bag. Google Cloud goes shopping.
By RepoJournal · Filed · About Google
JAX and Google Cloud both moved overnight, but it's the little housekeeping patches that will keep your CI green.
Overnight, googleapis/python-genai added server-initiated Video Understanding to the Interactions API [1], bringing ProcessingCallStep and ProcessingResultStep to streaming workflows. Meanwhile, Google Cloud onboarded two new client libraries: google-cloud-sql [2] and BigLake Hive v1 [3], both landing in google-cloud-python. Back in JAX, hawkinsp re-added the matplotlib version constraint after discovering "There is no 3.15 matplotlib wheel on pypi yet" [4], which will save you from broken docs builds. The team also deleted a sparsecore gather test because the SC compiler "is making progress supporting more gathers" [5]. For TPU users, Pallas Mosaic got a lowering rule for lax.approx_max_k [6], a proper top_k implementation that should speed up approximate max-k workloads. Two Python-genai PRs cover the same feature [1][7]; both are worth eyeballing. All told, 29 commits and 26 PRs across the three repos, so nothing is blocked.
Action items
- → Update python-genai to include Video Understanding support if you use Interactions API googleapis/python-genai [immediate]
- → Review Pallas Mosaic lowering for approx_max_k if you run on TPU google/jax [plan]
- → Onboard google-cloud-sql and BigLake Hive v1 into your dependency management googleapis/google-cloud-python [plan]
References
- [1] feat: add Video Understanding support to the Interactions API ↗ googleapis/python-genai
- [2] feat(google/cloud/sql): onboard google-cloud-sql ↗ googleapis/google-cloud-python
- [3] feat(google/cloud/biglake/hive/v1): onboard a new library ↗ googleapis/google-cloud-python
- [4] Readd version constraint on matplotlib. ↗ google/jax
- [5] Remove test test_sparsecore_unsupported_gather google/jax
- [6] [Mosaic TPU] Add lowering rule for `lax.approx_max_k` in Pallas Mosaic. First reduce the reduction dimension to some number of bins, then do top_k. ↗ google/jax
- [7] feat: add Video Understanding support to the Interactions API googleapis/python-genai