Who contributed to Google on July 12, 2026?

4 developers shipped this update, including copybara-service, Yash Katariya, Google Team Member, and Ayush Agrawal.

What were the notable Google updates?

Bring back SimpleNamespace export of checkpoint policies to fix typing failures, Convert out_shardings to physical shardings in presence of extended dtypes. The same happens for in_shardings. Fixes https://github.com/jax-ml/jax/issues/39089, and Convert out_shardings to physical shardings in presence of extended dtypes. The same happens for in_shardings. Fixes https://github.com/jax-ml/jax/issues/39089.

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 FIXES TYPING FAILURES AND SHARDING BUGS ACROSS CHECKPOINT AND EXTENDED DTYPE PIPELINES

By RepoJournal · Filed 06:57 UTC on July 12, 2026 · About Google

4 people shipped this

copybara-service @copybara-service 4 cited

Yash Katariya @yashk2810 1 cited

Google Team Member @google-genai-bot 1 cited

Ayush Agrawal @speedstorm1 1 cited

JAX shipped critical fixes overnight for checkpoint policy exports and extended dtype sharding, addressing both typing failures and physical sharding conversion issues that were breaking production inference.

The most urgent fix brings back SimpleNamespace export of checkpoint policies [1], resolving typing failures that were cascading through downstream code. In parallel, JAX converted out_shardings to physical shardings in the presence of extended dtypes [2] [3], fixing a critical issue where sharding metadata wasn't translating correctly for models using non-standard numeric types. These two fixes target the same class of problem: breakage in the checkpoint and sharding layers that affects anyone running distributed training or inference with modern model architectures. The python-genai team landed CodeMenderAgentConfig support [4] [5], expanding agent capabilities for code analysis workflows. Meanwhile, gemini-embedding-2 now populates per-modality prompt token counts in embedding responses [6] [7], giving you fine-grained cost tracking across multimodal embeddings. All three JAX fixes are shipping simultaneously, which means your next JAX upgrade will stabilize the entire distributed training surface.

Action items

→ Upgrade JAX immediately if using checkpoint policies or extended dtypes in distributed training google/jax [immediate]
→ Update python-genai to pull CodeMenderAgentConfig and embedding token tracking improvements googleapis/python-genai [plan]
→ Audit existing checkpoints for sharding mismatches if using extended dtypes google/jax [monitor]

References

[1] Bring back SimpleNamespace export of checkpoint policies to fix typing failures ↗ google/jax
[2] Convert out_shardings to physical shardings in presence of extended dtypes. The same happens for in_shardings. Fixes https://github.com/jax-ml/jax/issues/39089 google/jax
[3] Convert out_shardings to physical shardings in presence of extended dtypes. The same happens for in_shardings. Fixes https://github.com/jax-ml/jax/issues/39089 ↗ google/jax
[4] feat(interactions): add CodeMenderAgentConfig ↗ googleapis/python-genai
[5] feat(interactions): add CodeMenderAgentConfig googleapis/python-genai
[6] feat: Populate per-modality prompt token count in embedding responses for gemini-embedding-2 googleapis/python-genai
[7] feat: Populate per-modality prompt token count in embedding responses for gemini-embedding-2 ↗ googleapis/python-genai

Quick answers

What shipped in Google on July 12, 2026?: JAX shipped critical fixes overnight for checkpoint policy exports and extended dtype sharding, addressing both typing failures and physical sharding conversion issues that were breaking production inference. In total, 7 commits and 6 pull requests landed.
Who contributed to Google on July 12, 2026?: 4 developers shipped this update, including copybara-service, Yash Katariya, Google Team Member, and Ayush Agrawal.
What were the notable Google updates?: Bring back SimpleNamespace export of checkpoint policies to fix typing failures, Convert out_shardings to physical shardings in presence of extended dtypes. The same happens for in_shardings. Fixes https://github.com/jax-ml/jax/issues/39089, and Convert out_shardings to physical shardings in presence of extended dtypes. The same happens for in_shardings. Fixes https://github.com/jax-ml/jax/issues/39089.

JAX GPU KERNELS GET FASTER MEMORY ACCESS, BIGFRAMES UNLOCKS PANDAS-STYLE COLUMN INDEXING

JAX's Mosaic GPU backend shipped support for specialized matrix load/store instructions that will cut memory bottlenecks on supported hardware, while BigFrames now lets you access DataFrame columns by position like you're working with pandas.

ai 73 shipped 1-min read

Google 2 days ago

JAX FINALIZES CORE API DEPRECATIONS WHILE FP8 MIXED PRECISION LANDS ON MOSAIC GPU

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.

ai 66 shipped 1-min read

Google 3 days ago

JAX FIXES CUSTOM_VJP3 AUTODIFF BUGS; PYTHON-GENAI BREAKS BACKWARD COMPATIBILITY

JAX shipped critical fixes to custom_vjp3 that were producing wrong-shaped outputs under vmap-of-jvp, while python-genai landed a breaking change removing three parameters and reshaping safety settings.

ai 72 shipped 1-min read

Google 4 days ago

JAX ADDS ONEAPI GPU RUNTIME AS PYTHON 3.14 FORCES GOOGLE-AUTH DEPENDENCY PURGE

JAX shipped OneAPI (SYCL) GPU support while google-auth spent the night locking down five critical dependencies for Python 3.14 compatibility.

ai 55 shipped 1-min read

— or follow the whole beat:

Elsewhere on the wire

Elixir & Phoenix about 5 hours ago

ELIXIR CLOSES COMPLETION AND OPERATOR SIGNATURE GAPS

The Elixir team shipped fixes for two long-standing IDE friction points: cursor completion after operators and proper type signatures for the ++ concatenation operator.

elixir 2 shipped 1-min read

Mobile platforms about 5 hours ago

SWIFT TIGHTENS AVAILABILITY DIAGNOSTICS AS FLUTTER TRACKS SKIA UPDATES

Swift's compiler now correctly diagnoses uses of unavailable API members in universally unavailable extensions, closing a gap that let invalid code slip through.

mobile 8 shipped 1-min read

@dotnet about 5 hours ago

ANDROID SSL VALIDATION BYPASS PATCHED; WASM JIT ALIGNMENT FIXES BLOCKER

SslStream on Android was ignoring platform certificate pinning entirely, bypassing network_security_config.xml and allowing rejected certificates through.

dotnet 28 shipped 1-min read

@spring-projects about 5 hours ago

SPRING-AI FIXES SILENT DATA LOSS IN BATCH EMBEDDING STRATEGY

A critical bug in TokenCountBatchingStrategy was dropping duplicate documents during batch processing, silently returning fewer embeddings than requested.

java 9 shipped 1-min read