RepoJournal
Google

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 ADDS FP6 SUPPORT END-TO-END, PYTHON-GENAI SHIPS AGENT CONFIG

By RepoJournal · Filed · About Google

JAX landed full float6 format support across CPU and GPU backends, while the AI SDK gains AntigravityAgentConfig and a trigger resource in v2.12.0.

The JAX team shipped comprehensive FP6 support [1] spanning IFRT dtype definitions, PJRT C API bindings bumped to API version 114, and Python-level exposure of both float6_e2m3fn and float6_e3m2fn formats. This lands infrastructure for the memory-efficient training and inference everyone's been waiting for. On the experimental side, Mosaic GPU got multi-device barrier kernel support via XLA runtime APIs [2], and Pallas now has VJP rules for einshape [3], unblocking more gradient compute patterns. Google-cloud-python initiated PEP 0810 lazy imports for gapic v1 [5], deferring grpcio and other heavy C-extensions to reduce init latency and peak memory. The python-genai SDK bumped to v2.12.0 [4] with AntigravityAgentConfig additions and a new Triggers resource, while google-cloud-pubsub, google-cloud-spanner, and google-cloud-bigtable all regenerated against updated googleapis definitions [6] [7] [8].

Action items

References

  1. [1] Add end-to-end support for FP6 formats (float6_e2m3fn and float6_e3m2fn) across CPU and GPU. ↗ google/jax
  2. [2] [Mosaic:GPU] Use XLA runtime API to run multi-device barrier kernel ↗ google/jax
  3. [3] [pallas] VJP rule for the einshape primitive ↗ google/jax
  4. [4] v2.12.0 ↗ googleapis/python-genai
  5. [5] feat: lazy imports gapic v1 (#17673) googleapis/google-cloud-python
  6. [6] chore: regenerate google-cloud-pubsub and google-cloud-spanner ↗ googleapis/google-cloud-python
  7. [7] chore: regenerate google-cloud-bigtable ↗ googleapis/google-cloud-python
  8. [8] feat: update googleapis and regenerate ↗ googleapis/google-cloud-python

Quick answers

What shipped in Google on July 16, 2026?
JAX landed full float6 format support across CPU and GPU backends, while the AI SDK gains AntigravityAgentConfig and a trigger resource in v2.12.0. In total, 45 commits, 42 pull requests, and 1 releases landed.
Who contributed to Google on July 16, 2026?
8 developers shipped this update, including copybara-service, mattjj, release-please[bot], release-please, Annie Luc, Heba Alazzeh, parthea, and noahdietz.
What were the notable Google updates?
Add end-to-end support for FP6 formats (float6_e2m3fn and float6_e3m2fn) across CPU and GPU, [Mosaic:GPU] Use XLA runtime API to run multi-device barrier kernel, and [pallas] VJP rule for the einshape primitive.

More from Google

Daily updates, in your inbox

Follow Google

JAX, the GenAI SDK, and the Cloud libs — Google's open source layer We'll email you a link to confirm first.

Free. Confirm via email. Unsubscribe in one click.

— or follow the whole beat:

Elsewhere on the wire