feat(jax): support neural networks #4156
92.52% of diff hit (target 83.41%)
View this Pull Request on Codecov
92.52% of diff hit (target 83.41%)
Annotations
Check warning on line 18 in deepmd/backend/jax.py
codecov / codecov/patch
deepmd/backend/jax.py#L18
Added line #L18 was not covered by tests
Check warning on line 22 in deepmd/backend/jax.py
codecov / codecov/patch
deepmd/backend/jax.py#L22
Added line #L22 was not covered by tests
Check warning on line 25 in deepmd/backend/jax.py
codecov / codecov/patch
deepmd/backend/jax.py#L25
Added line #L25 was not covered by tests
Check warning on line 66 in deepmd/backend/jax.py
codecov / codecov/patch
deepmd/backend/jax.py#L66
Added line #L66 was not covered by tests
Check warning on line 77 in deepmd/backend/jax.py
codecov / codecov/patch
deepmd/backend/jax.py#L77
Added line #L77 was not covered by tests
Check warning on line 88 in deepmd/backend/jax.py
codecov / codecov/patch
deepmd/backend/jax.py#L88
Added line #L88 was not covered by tests
Check warning on line 99 in deepmd/backend/jax.py
codecov / codecov/patch
deepmd/backend/jax.py#L99
Added line #L99 was not covered by tests
Check warning on line 110 in deepmd/backend/jax.py
codecov / codecov/patch
deepmd/backend/jax.py#L110
Added line #L110 was not covered by tests