$ the-wire · showcase
JAX WIRES INTEL GPU SUPPORT, FIXES MOSAIC ASSEMBLY BUGS
By RepoJournal · Filed · About Google · Composed from the cited sources · methodology
10 people shipped this
copybara-service
@copybara-service
2 cited
Chalmer Lowe
@chalmerlowe
2 cited
mini-goel
@mini-goel
1 cited
Dimitar (Mitko) Asenov
@dimitar-asenov
1 cited
shuoweil
@shuoweil
1 cited
hebaalazzeh
@hebaalazzeh
1 cited
Allan Renucci
@allanrenucci
release-please[bot]
@release-please[bot]
Ayush Agrawal
@speedstorm1
parthea
@parthea
JAX landed native random number generation on Intel GPUs while fixing critical debugging directives in Mosaic GPU inline assembly, signaling accelerated support across vendor platforms.
The ThreeFry2x32 PRNG GPU kernel for Intel GPUs shipped in full [1], wiring oneAPI through registration, lowering, and wheel packaging to enable native RNG on Intel hardware. This complements ongoing platform work that included cleanup of obsolete version guards [2]. In parallel, Mosaic GPU fixed a serious debugging bug where `.loc` directives inside inline assembly blocks were incorrectly incremented on every line, pointing to wrong source locations [3]. The team also prepped for upcoming deprecation of `pl.core_map` by reducing test usages [4]. Over in the cloud SDK, google-cloud-python landed three high-impact fixes: test harness improvements for parallel CI output [5], brittleness fixes for `google-api-core` metadata changes in pubsub tests [6], and a namespace package resolution fix in the profiler that now handles PEP 420 packages without `__init__.py` [8]. BigQuery introduced `PendingDeprecationWarning` on `to_dataframe` and `to_arrow` methods to guide developers toward optimized `pandas-gbq` entry points [7].
Action items
- → Review Intel GPU kernel integration if shipping JAX on Xe hardware google/jax [plan]
- → Test profiler on namespace packages if you maintain implicit package structures googleapis/google-cloud-python [monitor]
- → Plan migration from to_dataframe/to_arrow to pandas-gbq entry points googleapis/google-cloud-python [plan]
References
- [1] [oneAPI] Add ThreeFry2x32 PRNG GPU kernel for Intel GPUs ↗ google/jax
- [2] [NFC] Remove no-longer needed version guards ↗ google/jax
- [3] [Mosaic GPU] Fix incorrect `.loc` directives inside inline asm blocks google/jax ↗
- [4] [pallas:mosaic] Reduced the number of usages of `pl.core_map` in tests ↗ google/jax
- [5] chore(ci): revamp parallel test outputs clarity and conciseness (#18002) ↗ googleapis/google-cloud-python
- [6] tests: handle internal metadata changes in google-api-core (#18037) ↗ googleapis/google-cloud-python
- [7] feat: add PendingDeprecationWarning for to_dataframe and to_arrow conversion methods ↗ googleapis/google-cloud-python
- [8] fix(profiler): support namespace packages without __init__.py and filter sample dirs ↗ googleapis/google-cloud-python
