Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
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
feat(jax): support neural networks #4156
feat(jax): support neural networks #4156
Changes from all commits
64df179
b8a33b1
a89f1eb
File filter
Filter by extension
Conversations
Jump to
There are no files selected for viewing
Check warning on line 18 in deepmd/backend/jax.py
Codecov / codecov/patch
deepmd/backend/jax.py#L18
Check warning on line 22 in deepmd/backend/jax.py
Codecov / codecov/patch
deepmd/backend/jax.py#L22
Check warning on line 25 in deepmd/backend/jax.py
Codecov / codecov/patch
deepmd/backend/jax.py#L25
Check warning on line 66 in deepmd/backend/jax.py
Codecov / codecov/patch
deepmd/backend/jax.py#L66
Check warning on line 77 in deepmd/backend/jax.py
Codecov / codecov/patch
deepmd/backend/jax.py#L77
Check warning on line 88 in deepmd/backend/jax.py
Codecov / codecov/patch
deepmd/backend/jax.py#L88
Check warning on line 99 in deepmd/backend/jax.py
Codecov / codecov/patch
deepmd/backend/jax.py#L99
Check warning on line 110 in deepmd/backend/jax.py
Codecov / codecov/patch
deepmd/backend/jax.py#L110
Check notice
Code scanning / CodeQL
Statement has no effect Note
Check notice
Code scanning / CodeQL
Statement has no effect Note