Skip to content
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.

Sign up for GitHub

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

Jump to bottom

Prototype SPMD using local process data #8842

Open
pgmoka opened this issue Mar 17, 2025 · 0 comments
Open

Prototype SPMD using local process data #8842

pgmoka opened this issue Mar 17, 2025 · 0 comments
Assignees
@pgmoka
Labels
enhancement New feature or request SPMD / Distributed

Comments

@pgmoka
Copy link
Collaborator

pgmoka commented Mar 17, 2025

As a sub-item for enabling 2D sharding with minibatch=True, we want to create a protoype of SPMD that uses local batching. This will hopefully be the beginning of an implementation that might substitute the current workaround.
@pgmoka pgmoka self-assigned this Mar 17, 2025
@ysiraichi ysiraichi added the enhancement New feature or request label Mar 18, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@pgmoka pgmoka

Labels
enhancement New feature or request SPMD / Distributed
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@ysiraichi @pgmoka