LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
jaxamg/__init__.py
jaxamg/_amgx.cc
jaxamg/_amgx.pyi
jaxamg/_amgx_solvers.h
jaxamg/_amgx_utils.h
jaxamg/_ext.py
jaxamg/_version.py
jaxamg/cache.py
jaxamg/config.py
jaxamg/jaxamg.py
jaxamg/matrices.py
jaxamg/mpi_utils.py
jaxamg/preconditioners.py
jaxamg/py.typed
jaxamg/sparsity.py
jaxamg/utils.py
jaxamg.egg-info/PKG-INFO
jaxamg.egg-info/SOURCES.txt
jaxamg.egg-info/dependency_links.txt
jaxamg.egg-info/requires.txt
jaxamg.egg-info/top_level.txt
tests/test_config.py
tests/test_gradient.py
tests/test_graph_coloring.py
tests/test_matrices.py
tests/test_matrix_formats.py
tests/test_mpi.py
tests/test_operator.py
tests/test_preconditioners.py
tests/test_solve.py
tests/test_sparsity_cache.py
tests/test_tracing.py