The Wire · Showcase
JAX STRIPS LEGACY CALL MACHINERY, PYTHON-GENAI OPENS DEFERRED SERVICE TIER
By RepoJournal · Filed · About Google
JAX eliminated its decades-old call primitives, streamlining the compiler stack and cutting code that downstream users now shoulder directly [ref:4].
The JAX team completed a major refactor by removing call_p and closed_call_p in favor of a unified eval_jaxpr_p primitive [1], which simplifies internal kernel dispatch but forces downstream projects to update their integration. Separately, JAX-Triton got AOT compilation for non-autotuned kernels [2], eliminating runtime PTX dependencies and slashing startup latency when loading compiled binaries. Over in python-genai, the deferred service tier is now publicly available on Vertex [3], opening cost-effective inference paths for teams willing to trade latency for lower burn. Meanwhile, pandas-gbq v0.35.1 shipped with an Arrow decoder for BigQuery read responses [4], giving pandas users faster data deserialization. The google-cloud-python generator also tightened up environment variable handling [5], reducing duplication across multi-service packages and laying groundwork for eventual upstream delegation to google-auth.
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
- → Review JAX downstream code for call_p references; stubs remain but deprecation is coming google/jax [plan]
- → Evaluate deferred service tier pricing on Vertex for workloads that can tolerate higher latency googleapis/python-genai [monitor]
- → Upgrade pandas-gbq to 0.35.1 if you read BigQuery data into pandas DataFrames googleapis/google-cloud-python [plan]
References
- [1] remove call_p and closed_call_p, just use eval_jaxpr_p ↗ google/jax
- [2] Implement AOT compilation for non-autotuned JAX-Triton kernels. google/jax
- [3] feat(api): make the deferred service tier publicly available on Vertex googleapis/python-genai
- [4] pandas-gbq: v0.35.1 ↗ googleapis/google-cloud-python
- [5] chore(generator): move `read_environment_variables` to `_compat.py.j2` ↗ googleapis/google-cloud-python