Skip to content

Added ViT training example#5455

Open
vfdev-5 wants to merge 1 commit into
mainfrom
docs-examples--vit
Open

Added ViT training example#5455
vfdev-5 wants to merge 1 commit into
mainfrom
docs-examples--vit

Conversation

@vfdev-5
Copy link
Copy Markdown
Collaborator

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

Description:

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

@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 2 times, most recently from d156add to 8985363 Compare May 5, 2026 16:16
@vfdev-5 vfdev-5 mentioned this pull request May 11, 2026
14 tasks
@vfdev-5 vfdev-5 force-pushed the docs-examples--vit branch from 8985363 to 5c951b7 Compare May 18, 2026 08:46
@vfdev-5 vfdev-5 force-pushed the docs-examples--vit branch from 5c951b7 to 029d59e Compare May 29, 2026 11:55
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants