PyPI page
Home page
Author:
Yi Liu, Xiantao Fan, Jian-Xun Wang
Summary:
JAX bindings for NVIDIA AmgX sparse linear solver with automatic differentiation support
Latest version:
0.1.3
Required dependencies:
jax
|
numpy
|
scipy
Optional dependencies:
black
|
jax
|
jaxamg
|
lineax
|
mkdocstrings
|
mpi4jax
|
mpi4py
|
mypy
|
pre-commit
|
pytest
|
pytest-mpi
|
pytest-mypy
|
ruff
|
zensical
Downloads last day:
88
Downloads last week:
109
Downloads last month:
348