Who contributed to Google on July 4, 2026?

1 developer shipped this update, including mattjj.

What were the notable Google updates?

[mutable-arrays] document Ref autodiff interactions, esp jax.vjp with_refs, [vjp3] add a fancy transpose rule for gather_p, for sparse gradient-ref updates, and [hijax] followups: better errors for missing hitype methods and device_put of hi values.

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 CUTS MEMORY OVERHEAD IN REVERSE-MODE DIFFERENTIATION WITH SPARSE GRADIENT UPDATES

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

1 person shipped this

mattjj @mattjj 4 cited

JAX's mutable arrays feature now handles autodiff with refs efficiently, transforming dense gradient computations into sparse in-place updates that cut memory pressure on large embedding models.

The JAX team documented critical interactions between Ref types and automatic differentiation [1], showing how jax.vjp with_refs accumulates gradients via += instead of overwriting them. This pairs with a new fancy transpose rule for gather operations [2] that replaces dense materialization with direct indexed updates: gather operations that look like embedding lookups (NumPy-style advanced indexing over leading axes) now become single in-place adds on the gradient ref instead of materializing a full dense array, scattering into it, and adding the result back. The practical impact hits immediately on production workloads. A follow-up on hijax type safety [3] hardens error handling for missing methods and device_put operations on higher-order JAX values, while the FFI docs now integrate hijax patterns with proper sharding semantics [4]. This is the autodiff efficiency win the framework has needed.

Action items

→ Review array_refs.py and .ipynb docs for jax.vjp with_refs patterns in your gradient computation google/jax [plan]
→ Audit gather operations in embedding layers for sparse gradient ref eligibility google/jax [plan]
→ Update hijax FFI callsites to match new sharding semantics google/jax [monitor]

References

[1] [mutable-arrays] document Ref autodiff interactions, esp jax.vjp with_refs ↗ google/jax
[2] [vjp3] add a fancy transpose rule for gather_p, for sparse gradient-ref updates ↗ google/jax
[3] [hijax] followups: better errors for missing hitype methods and device_put of hi values ↗ google/jax
[4] [hijax] update ffi docs with hijax, handle sharding ↗ google/jax

Quick answers

What shipped in Google on July 4, 2026?: JAX's mutable arrays feature now handles autodiff with refs efficiently, transforming dense gradient computations into sparse in-place updates that cut memory pressure on large embedding models. In total, 20 commits and 10 pull requests landed.
Who contributed to Google on July 4, 2026?: 1 developer shipped this update, including mattjj.
What were the notable Google updates?: [mutable-arrays] document Ref autodiff interactions, esp jax.vjp with_refs, [vjp3] add a fancy transpose rule for gather_p, for sparse gradient-ref updates, and [hijax] followups: better errors for missing hitype methods and device_put of hi values.

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.

elixir 10 shipped 1-min read

Want every project, not just this one?

Follow Google