LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
docs/source/conf.py
torchsparsegradutils/__init__.py
torchsparsegradutils/indexed_matmul.py
torchsparsegradutils/sparse_lstsq.py
torchsparsegradutils/sparse_matmul.py
torchsparsegradutils/sparse_solve.py
torchsparsegradutils.egg-info/PKG-INFO
torchsparsegradutils.egg-info/SOURCES.txt
torchsparsegradutils.egg-info/dependency_links.txt
torchsparsegradutils.egg-info/not-zip-safe
torchsparsegradutils.egg-info/requires.txt
torchsparsegradutils.egg-info/top_level.txt
torchsparsegradutils/benchmarks/__init__.py
torchsparsegradutils/benchmarks/batched_sparse_mm_rand.py
torchsparsegradutils/benchmarks/benchmark_suite.py
torchsparsegradutils/benchmarks/benchmark_utils.py
torchsparsegradutils/benchmarks/sparse_generic_solve_rand.py
torchsparsegradutils/benchmarks/sparse_generic_solve_suite.py
torchsparsegradutils/benchmarks/sparse_mm_rand.py
torchsparsegradutils/benchmarks/sparse_mm_suite.py
torchsparsegradutils/benchmarks/sparse_triangular_solve_rand.py
torchsparsegradutils/benchmarks/sparse_triangular_solve_suitesparse.py
torchsparsegradutils/benchmarks/visualize_benchmark_results.py
torchsparsegradutils/cupy/__init__.py
torchsparsegradutils/cupy/cupy_bindings.py
torchsparsegradutils/cupy/cupy_sparse_solve.py
torchsparsegradutils/distributions/__init__.py
torchsparsegradutils/distributions/constraints.py
torchsparsegradutils/distributions/sparse_multivariate_normal.py
torchsparsegradutils/encoders/__init__.py
torchsparsegradutils/encoders/pairwise_encoder.py
torchsparsegradutils/encoders/pairwise_voxel_encoder.py
torchsparsegradutils/jax/__init__.py
torchsparsegradutils/jax/jax_bindings.py
torchsparsegradutils/jax/jax_sparse_solve.py
torchsparsegradutils/tests/__init__.py
torchsparsegradutils/tests/conftest.py
torchsparsegradutils/tests/test_bicgstab.py
torchsparsegradutils/tests/test_config.py
torchsparsegradutils/tests/test_cupy_bindings.py
torchsparsegradutils/tests/test_cupy_sparse_solve.py
torchsparsegradutils/tests/test_dist_stats_helpers.py
torchsparsegradutils/tests/test_distributions.py
torchsparsegradutils/tests/test_doctests.py
torchsparsegradutils/tests/test_encoders.py
torchsparsegradutils/tests/test_indexed_matmul.py
torchsparsegradutils/tests/test_integration_pairwise_sparse_mvn.py
torchsparsegradutils/tests/test_jax_bindings.py
torchsparsegradutils/tests/test_jax_sparse_solve.py
torchsparsegradutils/tests/test_linear_cg.py
torchsparsegradutils/tests/test_lsmr.py
torchsparsegradutils/tests/test_minres.py
torchsparsegradutils/tests/test_quickstart_guide.py
torchsparsegradutils/tests/test_random.py
torchsparsegradutils/tests/test_sparse_lstsq.py
torchsparsegradutils/tests/test_sparse_matmul.py
torchsparsegradutils/tests/test_sparse_solve.py
torchsparsegradutils/tests/test_sparse_triangular_solve.py
torchsparsegradutils/tests/test_utils.py
torchsparsegradutils/tests/test_params/czyx_shifts.yaml
torchsparsegradutils/tests/test_params/pairwise_coo_indices.yaml
torchsparsegradutils/tests/test_params/xyz_coords.yaml
torchsparsegradutils/utils/__init__.py
torchsparsegradutils/utils/bicgstab.py
torchsparsegradutils/utils/dist_stats_helpers.py
torchsparsegradutils/utils/linear_cg.py
torchsparsegradutils/utils/lsmr.py
torchsparsegradutils/utils/minres.py
torchsparsegradutils/utils/random_sparse.py
torchsparsegradutils/utils/utils.py