Who contributed to Google on July 3, 2026?

2 developers shipped this update, including copybara-service[bot] and nbayati.

What were the notable Google updates?

Add is_stable flag to jax.lax.top_k, [Pallas][Triton] Reapply the missing code from "Added gpu_info and removed gpu device fallback from pallas_call_lowering" PR, and [Mosaic GPU] Add support for MXFP4 MMA with block_size=16.

Google

JAX, the GenAI SDK, and the Cloud libs - Google's open source layer

Pick a date

Topics: AI / ML Python Full archive →

The Wire · Showcase

JAX SHIPS STABLE SORT KNOB, GOOGLE CLOUD AUTH STRIPS PYOPENSSL

By RepoJournal · Filed 06:03 UTC on July 3, 2026 · About Google

2 people shipped this

copybara-service[bot] @copybara-service[bot] 2 cited

nbayati @nbayati 2 cited

JAX's top_k now lets you trade sort stability for speed, while google-cloud-python rips out pyOpenSSL for native crypto in the biggest auth overhaul this quarter.

JAX landed a performance lever on jax.lax.top_k [1] with a new is_stable flag defaulting to True, letting users opt into faster unstable sorts when tie-breaking order doesn't matter. That's paired with heavyweight GPU work: Mosaic GPU now handles MXFP4 matrix multiply at block_size=16 [3], unlocking Blackwell hardware capabilities that were previously locked to block_size=32, and Pallas reapplied critical missing code [2] from a prior GPU device handling refactor. google-cloud-python shipped a security-first auth rewrite [5] yanking pyOpenSSL entirely in favor of standard library ssl and cryptography, plus it bakes in RAM-only key handling via Linux memfd_create to keep secrets off disk entirely. The same auth module now catches PermissionErrors on workload certificate reads [6] instead of hanging for 30 seconds in sandboxed environments like snap packages. JAX also reorganized flattree internals [4] to make unpacking and filtering commute, a necessary fix for scan JVP operations that filter tangent zeros.

Action items

→ Audit your jax.lax.top_k calls - benchmark against is_stable=False if sort order variance is tolerable google/jax [plan]
→ Plan google-cloud-python auth upgrade before next service deploy - pyOpenSSL removal is breaking googleapis/google-cloud-python [plan]
→ Test workload auth in sandboxed environments after upgrading - PermissionError handling prevents startup hangs googleapis/google-cloud-python [monitor]

References

[1] Add is_stable flag to jax.lax.top_k google/jax
[2] [Pallas][Triton] Reapply the missing code from "Added gpu_info and removed gpu device fallback from pallas_call_lowering" PR google/jax
[3] [Mosaic GPU] Add support for MXFP4 MMA with block_size=16 ↗ google/jax
[4] Reorganize flattree implementation. ↗ google/jax
[5] refactor(auth): replace pyOpenSSL with standard ssl and cryptography ↗ googleapis/google-cloud-python
[6] fix(auth): handle PermissionError on workload certificates to avoid startup hang and crash ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on July 3, 2026?: JAX's top_k now lets you trade sort stability for speed, while google-cloud-python rips out pyOpenSSL for native crypto in the biggest auth overhaul this quarter. In total, 22 commits and 20 pull requests landed.
Who contributed to Google on July 3, 2026?: 2 developers shipped this update, including copybara-service[bot] and nbayati.
What were the notable Google updates?: Add is_stable flag to jax.lax.top_k, [Pallas][Triton] Reapply the missing code from "Added gpu_info and removed gpu device fallback from pallas_call_lowering" PR, and [Mosaic GPU] Add support for MXFP4 MMA with block_size=16.

JAX SYNCS XLA COMPILER STACK

JAX updated its XLA dependency to pull in the latest compiler improvements and optimizations.

ai 1 shipped 1-min read

Google 2 days ago

JAX OVERHAULS AUTODIFF INTERNALS WITH CUSTOM_VJP3, EXPANDS MESH API

JAX shipped a major redesign of custom differentiation rules that simplifies how you define vector-Jacobian products, while simultaneously opening up mesh operations to the public API.

ai 24 shipped 1-min read

Google 3 days ago

JAX FIXES REMAT3 CONSTANT HANDLING CRASH, GOOGLE CLOUD PYTHON CENTRALIZES REST TRANSCODING

Matthew Johnson patched a critical remat3 regression where closed-over constants were silently dropped, causing immediate crashes in production rematerialization pipelines [ref:1].

ai 50 shipped 1-min read

Google 4 days ago

JAX TIGHTENS XLA OPTIMIZATION CONTROLS, GOOGLE CLOUD PYTHON SETS UP FEATURE GATES

JAX makes XLA optimization levels human-readable while Google Cloud Python client libraries prepare a unified feature gating system for observability.

ai 54 shipped 1-min read

Elsewhere on the wire

AI Agents about 21 hours ago

CLINE SHIPS FIVE CRITICAL AGENT FIXES AFTER END-TO-END TESTING BLITZ

Saoudrizwan just landed a cascade of fixes that prevent the agent loop from running wild, stop file reads from resolving to the wrong directory, and plug the compaction pipeline.

ai-agents 16 shipped 2-min read

Local LLMs about 21 hours ago

LLAMA.CPP SHIPS MINIMAX-M3 SPARSE ATTENTION, VLLM UNLOCKS HETEROGENEOUS PREFILL-DECODE

The inference stack gets a new model architecture and a fundamental shift in how prefill and decode can run on different hardware configurations.

+10

llms 125 shipped 1-min read

@CachyOS about 21 hours ago

GLIBC 2.43 REBUILD SHIPS, INSTALLER JUMPS TO C++26

CachyOS pushed glibc 2.43 through the pipeline while the CLI installer upgraded its C++ standard to C++26, signaling a major refresh across core infrastructure.

infra 97 shipped 1-min read

Elixir & Phoenix about 21 hours ago

ECTO FIXES SUBQUERY PLANNING IN JOINS, ELIXIR PATCHES IEX PARSING

Ecto's query planner now handles subqueries correctly in interpolated join filters, while Elixir fixes a syntax error that broke 'not in' as a continuation operator in IEx.