Skip to content

Add JAX training tutorial#1302

Open
mtsokol wants to merge 1 commit into
google:mainfrom
mtsokol:jax-training-docs
Open

Add JAX training tutorial#1302
mtsokol wants to merge 1 commit into
google:mainfrom
mtsokol:jax-training-docs

Conversation

@mtsokol
Copy link
Copy Markdown
Contributor

@mtsokol mtsokol commented May 11, 2026

This PR adds a guide on plugging Grain dataset in JAX training.


📚 Documentation preview 📚: https://google-grain--1302.org.readthedocs.build/

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant