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 FINALIZES CORE API DEPRECATIONS WHILE FP8 MIXED PRECISION LANDS ON MOSAIC GPU

By RepoJournal · Filed · About Google

JAX is cleaning house on jax.core APIs while shipping native mixed E4M3/E5M2 FP8 support that enterprises using Mosaic GPUs have been waiting for.

The jax.core deprecation finalization [1] closes a long-running cleanup that removes APIs marked for removal, clearing technical debt before the next major version. Simultaneously, Mosaic GPU gains native mixed-precision FP8 operand support [2] through independent .atype/.btype control in wgmma, enabling the exact training recipe used by NVIDIA's Transformer Engine where gradients stay in E5M2 for dynamic range while weights and activations use E4M3 for precision. On the Jaxpr front, a follow-up consolidation [3] pairs with a rollback revert [4], suggesting the core team is stabilizing the Jaxpr/ClosedJaxpr merge that's been churning through iterations. Elsewhere, python-genai shipped v2.11.0 [5] with environment_id network config and new retrieval result steps, while google-cloud-python landed async storage improvements [6] including optional checksum disabling for append performance and bigframes UDF transpiler support [7] for groupby operations.

One email a day. Unsubscribe in one click.

Action items

References

  1. [1] Finalize deprecation of several `jax.core` APIs ↗ google/jax
  2. [2] [Mosaic GPU] Support mixed E4M3/E5M2 FP8 operands in wgmma ↗ google/jax
  3. [3] Follow-up to Jaxpr/ClosedJaxpr merge ↗ google/jax
  4. [4] rollback of rollback of ClosedJaxpr/Jaxpr merge ↗ google/jax
  5. [5] v2.11.0 ↗ googleapis/python-genai
  6. [6] feat(storage): add option to disable checksums and improve robustness of full_object_checksum validation ↗ googleapis/google-cloud-python
  7. [7] feat(bigframes): Support groupby.agg/transform with udf transpiler ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on July 10, 2026?
JAX is cleaning house on jax.core APIs while shipping native mixed E4M3/E5M2 FP8 support that enterprises using Mosaic GPUs have been waiting for. In total, 35 commits, 30 pull requests, and 1 releases landed.
Who contributed to Google on July 10, 2026?
5 developers shipped this update, including copybara-service[bot], mcwitt, dougalm, chandra-siri, and TrevorBergeron.
What were the notable Google updates?
Finalize deprecation of several `jax.core` APIs, [Mosaic GPU] Support mixed E4M3/E5M2 FP8 operands in wgmma, and Follow-up to Jaxpr/ClosedJaxpr merge.

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?

We use privacy-friendly analytics (Google Analytics, IP-anonymized) to see which pages help readers. No ads, and we never sell your data. See our Privacy Policy.