Skip to content

Update JAX tutorial to reflect changes in device mesh creation and sharding strategies. - #288

Open
Davidnet wants to merge 1 commit into
jax-ml:mainfrom
Davidnet:davidnet-update-jax-llm-pretraining
Open

Update JAX tutorial to reflect changes in device mesh creation and sharding strategies.#288
Davidnet wants to merge 1 commit into
jax-ml:mainfrom
Davidnet:davidnet-update-jax-llm-pretraining

Update JAX tutorial to reflect changes in device mesh creation and sh…

7065eb8
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Dec 22, 2025 in 2s

✅ 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:

✅ 7065eb8 Author: @Davidnet <dav*******ozo​@me.com>

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

View more details on Google CLA