-
Notifications
You must be signed in to change notification settings - Fork 610
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
Add automatic parallelization JAX example #4973
Add automatic parallelization JAX example #4973
Conversation
Signed-off-by: Albert Wolant <awolant@nvidia.com>
Signed-off-by: Albert Wolant <awolant@nvidia.com>
Check out this pull request on See visual diffs & provide feedback on Jupyter Notebooks. Powered by ReviewNB |
Signed-off-by: Albert Wolant <awolant@nvidia.com>
Signed-off-by: Albert Wolant <awolant@nvidia.com>
Signed-off-by: Albert Wolant <awolant@nvidia.com>
Signed-off-by: Albert Wolant <awolant@nvidia.com>
Signed-off-by: Albert Wolant <awolant@nvidia.com>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This file is part of #4969
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This file is part of #4969
Signed-off-by: Albert Wolant <awolant@nvidia.com>
@@ -27,7 +27,7 @@ | |||
}, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
(single GPU training example)[jax-basic_example.ipynb]
Is it some kind of markup for Sphinx or sth?
Reply via ReviewNB
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Fixed
Signed-off-by: Albert Wolant <awolant@nvidia.com>
!build |
CI MESSAGE: [9229567]: BUILD STARTED |
CI MESSAGE: [9230402]: BUILD STARTED |
CI MESSAGE: [9229567]: BUILD PASSED |
CI MESSAGE: [9230402]: BUILD PASSED |
Adds example of automatic parallelization with JAX and DALI. Signed-off-by: Albert Wolant <awolant@nvidia.com>
Depends on #4969
Category:
New feature
Description:
Adds example of automatic parallelization with JAX and DALI.
Additional information:
Affected modules and functionalities:
JAX plugin docs.
Key points relevant for the review:
Is it understandable? Spelling and syntax?
Tests:
Checklist
Documentation
DALI team only
Requirements
REQ IDs: N/A
JIRA TASK: DALI-3558