RepoJournal
Google

Google

JAX, the GenAI SDK, and the Cloud libs โ€” Google's open source layer

Pick a date

  1. Fri 29 may
  2. Sat 30 may
  3. Sun 31 may
  4. Mon 1 jun
  5. Mon 8 jun
  6. Tue 9 jun
  7. Wed 10 jun
  8. Thu 11 jun
  9. Fri 12 jun
  10. Sat 13 jun
  11. Sun 14 jun
  12. Mon 15 jun
  13. Tue 16 jun
  14. Wed 17 jun
  15. Thu 18 jun
  16. Fri 19 jun
  17. Sat 27 jun
  18. Sun 28 jun
  19. Mon 29 jun
  20. Tue 30 jun
  21. Wed 1 jul
  22. Thu 2 jul
  23. Fri 3 jul
  24. Sat 4 jul
  25. Sun 5 jul
  26. Mon 6 jul
  27. Tue 7 jul
  28. Wed 8 jul
  29. Thu 9 jul
  30. Today 10 jul

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.

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

Related across the cluster

For your repos

The showcase is a teaser.
Your wire is the product.

Same engine. Different stack. Below: what changes when the wire is yours.

Showcase wire

  • โ€ข14 famous open source orgs
  • โ€ขOne wire per day
  • โ€ขPublic, generic
  • โ€ขRead on the web, when you remember

Your wire

  • โ†’Up to 1,500 of your repos - orgs, deps, vendors
  • โ†’Morning and evening briefs
  • โ†’Action items routed to your team
  • โ†’Slack delivery, email, breaking-news CVE alerts

Want a hands-on demo first? Ask a current user for an invite link.