LICENSE
MANIFEST.in
README.md
TODO.md
pyproject.toml
setup.py
csrc/spsolve/spsolve.cpp
tests/test_amg_preconditioner.py
tests/test_amgx_backend.py
tests/test_batch_solve.py
tests/test_complex_support.py
tests/test_cudss_autodetect.py
tests/test_distributed_cg_shard_multiprocess.py
tests/test_distributed_gather_owned_to_global.py
tests/test_distributed_krylov_shard_multiprocess.py
tests/test_distributed_precond_shard_multiprocess.py
tests/test_dsparse_batch_shard_multiprocess.py
tests/test_dsparse_eigsh_multiprocess.py
tests/test_dsparse_elementwise_multiprocess.py
tests/test_dsparse_io_multiprocess.py
tests/test_dsparse_io_single_process.py
tests/test_dsparse_logdet_multiprocess.py
tests/test_dsparse_reductions_multiprocess.py
tests/test_dsparse_tensor_mirror_props_multiprocess.py
tests/test_dsparse_tensor_to_device_dtype_multiprocess.py
tests/test_dsparse_topology_multiprocess.py
tests/test_lobpcg_eigsh.py
tests/test_partition_hilbert.py
tests/test_pyamg_backend.py
tests/test_solve_api.py
tests/test_solver_cache.py
tests/test_solver_config.py
tests/test_solver_config_presets.py
tests/test_sparse_tensor.py
tests/test_sparse_tensor_extract_partition.py
tests/test_sparse_tensor_factories.py
tests/test_sparse_tensor_list.py
tests/test_spsolve.py
tests/test_unified_solve_dispatches_to_dsparse.py
torch_sla/__init__.py
torch_sla/batch_solve.py
torch_sla/benchmark.py
torch_sla/datasets.py
torch_sla/det.py
torch_sla/io.py
torch_sla/linear_solve.py
torch_sla/nonlinear_solve.py
torch_sla/partition.py
torch_sla/presets.py
torch_sla/solve.py
torch_sla/solver_cache.py
torch_sla/backends/__init__.py
torch_sla/backends/amgx_backend.py
torch_sla/backends/cupy_backend.py
torch_sla/backends/nvmath_backend.py
torch_sla/backends/pyamg_backend.py
torch_sla/backends/pytorch_backend.py
torch_sla/backends/scipy_backend.py
torch_sla/backends/torch_spsolve_backend.py
torch_sla/distributed/__init__.py
torch_sla/distributed/collectives.py
torch_sla/distributed/core.py
torch_sla/distributed/eigsh.py
torch_sla/distributed/matvec.py
torch_sla/distributed/solve.py
torch_sla/sparse_tensor/__init__.py
torch_sla/sparse_tensor/autograd.py
torch_sla/sparse_tensor/convert.py
torch_sla/sparse_tensor/core.py
torch_sla/sparse_tensor/graph.py
torch_sla/sparse_tensor/linalg.py
torch_sla/sparse_tensor/list.py
torch_sla/sparse_tensor/matmul.py
torch_sla/sparse_tensor/ops.py
torch_sla/sparse_tensor/reductions.py
torch_sla/sparse_tensor/structural.py
torch_sla/sparse_tensor/utils.py
torch_sla/sparse_tensor/viz.py