Skip to content

docs: new sharding docs #1370

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
wants to merge 5 commits into
base: main
Choose a base branch
from
Draft

docs: new sharding docs #1370

wants to merge 5 commits into from

Conversation

avik-pal
Copy link
Collaborator

@avik-pal avik-pal commented Jun 4, 2025

No description provided.

@avik-pal avik-pal force-pushed the ap/sharding_docs branch from a9ac5dd to 09a1ff7 Compare June 4, 2025 19:14
@giordano giordano added documentation Improvements or additions to documentation sharding labels Jun 12, 2025

### 4-way Batch & 2-way Model Parallelism

## Related links
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'm not sure this is relevant here since that's about hardware rather than software (although it's some hardware which is very related to XLA), but recently I came across this very good overview about TPUs: https://henryhmko.github.io/posts/tpu/tpu.html

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
documentation Improvements or additions to documentation sharding
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants