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

Topics: AI / ML Python Full archive →

The Wire · Showcase

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

By RepoJournal · Filed · About Google

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

The JAX team completed a major refactor by removing call_p and closed_call_p in favor of a unified eval_jaxpr_p primitive [1], which simplifies internal kernel dispatch but forces downstream projects to update their integration. Separately, JAX-Triton got AOT compilation for non-autotuned kernels [2], eliminating runtime PTX dependencies and slashing startup latency when loading compiled binaries. Over in python-genai, the deferred service tier is now publicly available on Vertex [3], opening cost-effective inference paths for teams willing to trade latency for lower burn. Meanwhile, pandas-gbq v0.35.1 shipped with an Arrow decoder for BigQuery read responses [4], giving pandas users faster data deserialization. The google-cloud-python generator also tightened up environment variable handling [5], reducing duplication across multi-service packages and laying groundwork for eventual upstream delegation to google-auth.

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] remove call_p and closed_call_p, just use eval_jaxpr_p ↗ google/jax
  2. [2] Implement AOT compilation for non-autotuned JAX-Triton kernels. google/jax
  3. [3] feat(api): make the deferred service tier publicly available on Vertex googleapis/python-genai
  4. [4] pandas-gbq: v0.35.1 ↗ googleapis/google-cloud-python
  5. [5] chore(generator): move `read_environment_variables` to `_compat.py.j2` ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on August 7, 2026?
JAX eliminated its decades-old call primitives, streamlining the compiler stack and cutting code that downstream users now shoulder directly [ref:4]. In total, 28 commits, 25 pull requests, and 1 releases landed.
Who contributed to Google on August 7, 2026?
12 developers shipped this update, including Yash Katariya, Will Froom, Sergei Lebedev, mattjj, Eusebio Durán Montaña, copybara-service, Google Team Member, and Mark Daoust, and 4 more.
What were the notable Google updates?
remove call_p and closed_call_p, just use eval_jaxpr_p, Implement AOT compilation for non-autotuned JAX-Triton kernels, and feat(api): make the deferred service tier publicly available on Vertex.

More from Google

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

Want every project, not just this one?