Skip to content

Fix: Modernize sharding detection and error messages for JAX 0.8.2 compatibility (#422)#423

Open
Ashutosh0x wants to merge 2 commits into
google-deepmind:mainfrom
Ashutosh0x:fix/jax-0.8.2-compat
Open

Fix: Modernize sharding detection and error messages for JAX 0.8.2 compatibility (#422)#423
Ashutosh0x wants to merge 2 commits into
google-deepmind:mainfrom
Ashutosh0x:fix/jax-0.8.2-compat

fix: address review comments (compatibility and style adjustments)

ba1e5fc
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Jan 6, 2026 in 1s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

ba1e5fc Author: @Ashutosh0x <ashut**********ngh0x​@gmail.com>

(Only the first commit for a unique contributor is listed.)