RepoJournal
Google

Google

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

Keep up with Google in about 3 minutes a day: what actually shipped — the commits, pull requests, releases, and security advisories that matter.

One email a day. Unsubscribe in one click.

Pick a date

  1. Sat 11 jul
  2. Sun 12 jul
  3. Mon 13 jul
  4. Tue 14 jul
  5. Wed 15 jul
  6. Thu 16 jul
  7. Sat 18 jul
  8. Sun 19 jul
  9. Mon 20 jul
  10. Tue 21 jul
  11. Wed 22 jul
  12. Thu 23 jul
  13. Fri 24 jul
  14. Sat 25 jul
  15. Sun 26 jul
  16. Mon 27 jul
  17. Tue 28 jul
  18. Wed 29 jul
  19. Thu 30 jul
  20. Fri 31 jul
  21. Sat 1 aug
  22. Sun 2 aug
  23. Mon 3 aug
  24. Tue 4 aug
  25. Wed 5 aug
  26. Thu 6 aug
  27. Fri 7 aug
  28. Sat 8 aug
  29. Sun 9 aug
  30. Today 10 aug
Topics: AI / ML Python Full archive →

The Wire · Showcase

JAX FIXES ZERO-PROPAGATION BUG IN REVERSE-MODE AUTODIFF

By RepoJournal · Filed · About Google

mattjj copybara-service Junwhan Ahn
3 people shipped this
mattjj @mattjj 2 cited copybara-service @copybara-service 1 cited Junwhan Ahn @junwhanahn

JAX's core autodiff engine now correctly propagates zero tangents through remat operations, fixing a downstream Flax pattern that was leaking Tracers into production code.

The fix lands in jax.vjp and jax.linearize with a new in_nzs parameter, a tuple-tree of per-input nonzero-tangent flags that lets dead code paths stay dead [1]. This prevents the tracer leaks that were surfacing in certain gradient computations, a subtle but critical correctness issue for anyone chaining JAX's autodiff with complex control flow. The change is fully backward compatible: the returned callable still accepts tangent inputs corresponding to zeros, so existing code keeps working. In parallel, the team removed process_call as dead code [2], a cleanup that shrinks the autodiff surface and reduces maintenance burden. A separate quality-of-life fix caps array repr output at 1MiB to prevent logging overhead when debugging with large tensors [3].

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] [remat3][vjp3] propagate zeros through remat3 via jax.vjp/linearize ↗ google/jax
  2. [2] [cleanup] remove process_call, now dead code google/jax
  3. [3] Do not print the content of an array from `repr`/`str` if it's bigger than 1MiB ↗ google/jax

Quick answers

What shipped in Google on August 9, 2026?
JAX's core autodiff engine now correctly propagates zero tangents through remat operations, fixing a downstream Flax pattern that was leaking Tracers into production code. In total, 5 commits and 3 pull requests landed.
Who contributed to Google on August 9, 2026?
3 developers shipped this update, including mattjj, copybara-service, and Junwhan Ahn.
What were the notable Google updates?
[remat3][vjp3] propagate zeros through remat3 via jax.vjp/linearize, [cleanup] remove process_call, now dead code, and Do not print the content of an array from `repr`/`str` if it's bigger than 1MiB.

More from Google

Google Google

JAX REVERTS BREAKING CHANGE, SYNCS XLA UPSTREAM

A changelist that hit production got rolled back twice in 12 hours, signaling either a critical regression or a coordination miss that JAX caught before it spread.

copybara-service Jinliang Wei jax authors
ai 3 shipped
Google Google

JAX OVERHAULS REF SYSTEM AS GENAI SDK LOCKS IN JSON SCHEMA SUPPORT

JAX's reference layer just got a major surgery: pin control arguments are in, version guards are out, and Pallas GPU kernels gained conditional execution.

mattjj Yash Katariya vfdev-5 Allan Renucci +7
ai 75 shipped
Google Google

JAX STRIPS LEGACY CALL MACHINERY, PYTHON-GENAI OPENS DEFERRED SERVICE TIER

JAX eliminated its decades-old call primitives, streamlining the compiler stack and cutting code that downstream users now shoulder directly [ref:4].

Yash Katariya Will Froom Sergei Lebedev mattjj +8
ai 54 shipped
Google Google

JAX PINS PYTHON 3.14T, GENAI GAINS ROBOTICS MODEL AND ERROR HANDLING

JAX disabled profiler tests on free-threaded Python to unblock CI, while python-genai 2.17.0 ships the Gemini Robotics ER 2 Preview and fixes tool call limits.

Kanglan Tang copybara-service mattjj release-please[bot] +6
ai 90 shipped

Daily updates, in your inbox

Follow Google

Keep up with Google in about 3 minutes a day: what actually shipped — the commits, pull requests, releases, and security advisories that matter.

or

One email a day. Unsubscribe in one click. Read a past issue →

Elsewhere on the wire

AI Agents AI Agents

LANGCHAIN PATCHES PYDANTIC 2.14 BREAKS, CLINE KILLS YOLO MODE

LangChain shipped critical compatibility fixes for Pydantic 2.14 serialization bugs while Cline removed a cosmetic toggle that masked real approval confusion.

langchain-oss-model-profiles[bot] ccurme langchain-oss-model-profiles saoudrizwan +1
ai-agents 10 shipped
Local LLMs Local LLMs

OLLAMA SHIPS VISION SUPPORT ACROSS MLX AND QWEN3.5 AS LLAMA.CPP FIXES DISPATCH BUGS

Ollama's MLX runner now processes images natively, preserving prefix caching and speculative decoding while rejecting unsupported requests with clear errors.

jessegross Daniel Hiltgen github-actions[bot] allozaur +9
llms 157 shipped
CachyOS @CachyOS

CACHYOS LINUX 7.2-RC7 SHIPS WITH WIFI AND USB FIXES, SETTINGS 1.4.0 REMOVES NVIDIA DEPRECATED FLAG

CachyOS Linux 7.2-rc7-1 rebased upstream with critical fixes for MT76 wifi devices and USB initialization, while the settings package ditches legacy NVIDIA parameters.

Eric Naim ptr1337 Dharmik-Lathiya Vladislav Nepogodin +2
infra 73 shipped
Elixir & Phoenix Elixir & Phoenix

ELIXIR FIXES UTF-8 CORRUPTION IN MACRO.TO_STRING, ECTO CUTS COMPILE DEPS

A critical fix landed for Elixir's string handling that was silently corrupting UTF-8 sequences when converting C1 control characters, while Ecto shipped a change that trims unnecessary compile-time dependencies from schema derivations.

mnnekrashevich woylie
elixir 4 shipped
Want every project, not just this one?