Skip to content

WIP: pathways checkpointing experiments#1340

Closed
samos123 wants to merge 51 commits into
apple:mainfrom
samos123:pathways-array-serialization
Closed

WIP: pathways checkpointing experiments#1340
samos123 wants to merge 51 commits into
apple:mainfrom
samos123:pathways-array-serialization

Conversation

@samos123
Copy link
Copy Markdown
Contributor

No description provided.

class TensorstoreSpecModifier:
def __call__(self, spec: dict[str, Any], *, shard_infos: list[_ShardInfo]):
...
def __call__(self, spec: dict[str, Any], *, shard_infos: list[_ShardInfo]): ...
Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

i guess black did this

@samos123 samos123 changed the title WIP: send all jax device puts in parallel with pathways WIP: pathways checkpointing experiments Sep 12, 2025
@samos123 samos123 force-pushed the pathways-array-serialization branch from 2249843 to cf24047 Compare September 16, 2025 18:20
@github-actions
Copy link
Copy Markdown

This pull request has been automatically marked as stale because it has been inactive for 60 days. It will be closed in 7 days if no further activity occurs. If you would like to continue working on this, please remove the stale label or leave a comment.

@github-actions github-actions Bot added the stale label Dec 27, 2025
@github-actions
Copy link
Copy Markdown

github-actions Bot commented Jan 4, 2026

This pull request was closed because it has been inactive for more than 7 days since being marked as stale. Please feel free to reopen it if you would like to continue.

@github-actions github-actions Bot closed this Jan 4, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant