The Wire · Showcase
JAX PINS PYTHON 3.14T, GENAI GAINS ROBOTICS MODEL AND ERROR HANDLING
By RepoJournal · Filed · About Google
JAX disabled profiler tests on free-threaded Python to unblock CI, while python-genai 2.17.0 ships the Gemini Robotics ER 2 Preview and fixes tool call limits.
JAX's profiler test now skips on all free-threaded Python versions [1] after hitting timeout walls on Python 3.14t CI. The same release cycle saw heavy Mosaic GPU work land: a new b6x16_p32 MLIR dtype [2], unified Pallas and Mosaic compilation pipelines [3], and debug dumps for layout inference [4] via MOSAIC_GPU_DUMP_MLIR_PASSES. Python-genai 2.17.0 [5] adds the Gemini Robotics ER 2 Preview model and introduces TOO_MANY_TOOL_CALLS to the FinishReason enum [6], addressing a gap when models exhaust their tool budget. On the cloud-python side, a proto-plus message fix landed in api-core [7] that restores UUID auto-population by switching from identity checks to truthiness checks, unblocking downstream libraries. The import profiler script also got faster [8] by calculating codebase statistics only once instead of on every iteration across 22 test runs. Google-cloud-python bumped api-core to 2.28.0 [9] to consolidate version parsing logic and reduce duplication.
One email a day. Unsubscribe in one click.
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. Read a past issue →
Action items
- → Update google-cloud-python to api-core 2.28.0 before next release cycle googleapis/google-cloud-python [plan]
- → Pin python-genai to 2.17.0 if you use tool calling with many models googleapis/python-genai [plan]
- → Monitor JAX free-threaded Python support for 3.14+ stability google/jax [monitor]
References
- [1] [CI] Skip profiler tests on all free-threaded Python versions google/jax
- [2] [Mosaic GPU] Define the `b6x16_p32` MLIR dtype. ↗ google/jax
- [3] [Pallas/Mosaic GPU][NFC] Deduplicate Pallas and Mosaic GPU compilation pipelines. ↗ google/jax
- [4] [Pallas/Mosaic GPU] Allow dumping module before and after layout inference. ↗ google/jax
- [5] v2.17.0 ↗ googleapis/python-genai
- [6] feat: Add TOO_MANY_TOOL_CALLS to FinishReason enum. googleapis/python-genai
- [7] fix(api-core): use truthiness check in setup_request_id to support proto-plus messages (#18000) googleapis/google-cloud-python
- [8] fix(ci): skip loaded lines calculation on iterations after the first ↗ googleapis/google-cloud-python
- [9] chore: bump google-api-core to 2.28.0 ↗ googleapis/google-cloud-python