Skip to content

Added ViT training example#5455

Open
vfdev-5 wants to merge 2 commits intomainfrom
docs-examples--vit
Open

Added ViT training example#5455
vfdev-5 wants to merge 2 commits intomainfrom
docs-examples--vit

Conversation

@vfdev-5
Copy link
Copy Markdown
Collaborator

@vfdev-5 vfdev-5 commented May 5, 2026

!!! Based on #5434

Description:

  • ported ViT training example from Jax AI Stack
    • new: explicit sharding support

@review-notebook-app
Copy link
Copy Markdown

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@vfdev-5 vfdev-5 changed the title Docs examples vit Added ViT training example May 5, 2026
@vfdev-5 vfdev-5 force-pushed the docs-examples--vit branch from c44bf8e to d156add Compare May 5, 2026 14:57
@vfdev-5 vfdev-5 force-pushed the docs-examples--vit branch from d156add to 8985363 Compare May 5, 2026 16:16
@vfdev-5 vfdev-5 mentioned this pull request May 11, 2026
12 tasks
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