.gitignore
.pre-commit-config.yaml
.readthedocs.yaml
LICENSE
README.md
pyproject.toml
pytest.ini
setup.cfg
setup.py
.github/ISSUE_TEMPLATE/---bug-report.md
.github/ISSUE_TEMPLATE/---documentation-examples.md
.github/ISSUE_TEMPLATE/---feature-request.md
.github/ISSUE_TEMPLATE/---other.md
.github/ISSUE_TEMPLATE/---refactor.md
.github/workflows/deploy.yml
.github/workflows/format_linter_enforcer.yml
.github/workflows/pull_request.yml
.github/workflows/push_to_main.yml
.github/workflows/run_test_suite.yml
cola/__init__.py
cola/annotations.py
cola/fns.py
cola/version.py
cola/backends/__init__.py
cola/backends/backends.py
cola/backends/jax_fns.py
cola/backends/np_fns.py
cola/backends/torch_fns.py
cola/linalg/__init__.py
cola/linalg/algorithm_base.py
cola/linalg/decompositions/__init__.py
cola/linalg/decompositions/arnoldi.py
cola/linalg/decompositions/decompositions.py
cola/linalg/decompositions/lanczos.py
cola/linalg/eig/__init__.py
cola/linalg/eig/eigs.py
cola/linalg/eig/iram.py
cola/linalg/eig/lobpcg.py
cola/linalg/eig/power_iteration.py
cola/linalg/inverse/__init__.py
cola/linalg/inverse/cg.py
cola/linalg/inverse/gmres.py
cola/linalg/inverse/inv.py
cola/linalg/inverse/pinv.py
cola/linalg/logdet/__init__.py
cola/linalg/logdet/logdet.py
cola/linalg/preconditioning/preconditioners.py
cola/linalg/svd/svd.py
cola/linalg/tbd/nullspace.py
cola/linalg/tbd/qr.py
cola/linalg/tbd/randomized_svd.py
cola/linalg/tbd/slq.py
cola/linalg/tbd/svrg.py
cola/linalg/trace/__init__.py
cola/linalg/trace/diag_trace.py
cola/linalg/trace/diagonal_estimation.py
cola/linalg/unary/__init__.py
cola/linalg/unary/unary.py
cola/ops/__init__.py
cola/ops/operator_base.py
cola/ops/operators.py
cola/utils/__init__.py
cola/utils/custom_autodiff.py
cola/utils/jax_tqdm.py
cola/utils/torch_tqdm.py
cola/utils/utils_for_tests.py
cola/utils/utils_linalg.py
cola_ml.egg-info/PKG-INFO
cola_ml.egg-info/SOURCES.txt
cola_ml.egg-info/dependency_links.txt
cola_ml.egg-info/requires.txt
cola_ml.egg-info/top_level.txt
docs/CoLA_Table1.png
docs/CoLA_Table2.png
docs/Makefile
docs/conf.py
docs/contributing.md
docs/index.rst
docs/requirements.txt
docs/_static/logo.svg
docs/_static/style.css
docs/_templates/layout.html
docs/notebooks/01_algorithms.ipynb
docs/notebooks/03_GPs.ipynb
docs/notebooks/04_Spectral_Clustering.ipynb
docs/notebooks/05_Boundary_Value_PDEs.ipynb
docs/notebooks/06_PDE_Eigenvalue_Problems.ipynb
docs/notebooks/07_second_order_optimization.ipynb
docs/notebooks/08_dispatch_functionality.ipynb
docs/notebooks/09_hessian_spectrum.ipynb
docs/notebooks/10_vmap_jit_grad.ipynb
docs/notebooks/About.ipynb
docs/notebooks/Adding_Linear_Operators.ipynb
docs/notebooks/Device_Allocation.ipynb
docs/notebooks/Linalg_operations_walkthrough.ipynb
docs/notebooks/Quick_Start.ipynb
docs/notebooks/_colab_preamble.ipynb
docs/notebooks/merge_nbs.sh
docs/notebooks/tricky_bits.ipynb
docs/notebooks/worksheet.ipynb
docs/notebooks/colabs/01_PCA.ipynb
docs/notebooks/colabs/01_algorithms.ipynb
docs/notebooks/colabs/02_Linear_Regression.ipynb
docs/notebooks/colabs/03_GPs.ipynb
docs/notebooks/colabs/04_Spectral_Clustering.ipynb
docs/notebooks/colabs/05_Boundary_Value_PDEs.ipynb
docs/notebooks/colabs/06_PDE_Eigenvalue_Problems.ipynb
docs/notebooks/colabs/07_second_order_optimization.ipynb
docs/notebooks/colabs/08_dispatch_functionality.ipynb
docs/notebooks/colabs/09_hessian_spectrum.ipynb
docs/notebooks/colabs/10_vmap_jit_grad.ipynb
docs/notebooks/colabs/About.ipynb
docs/notebooks/colabs/Adding_Linear_Operators.ipynb
docs/notebooks/colabs/Device_Allocation.ipynb
docs/notebooks/colabs/Examples_on_Basic_Functionality.ipynb
docs/notebooks/colabs/LinOpIntro.ipynb
docs/notebooks/colabs/Linalg_operations_walkthrough.ipynb
docs/notebooks/colabs/Lower_Precision.ipynb
docs/notebooks/colabs/Quick_Start.ipynb
docs/notebooks/colabs/all.ipynb
docs/notebooks/colabs/sharp_bits.ipynb
docs/notebooks/colabs/tricky_bits.ipynb
docs/notebooks/colabs/worksheet.ipynb
docs/package/cola.annotations.rst
docs/package/cola.decompositions.rst
docs/package/cola.fns.rst
docs/package/cola.linalg.rst
docs/package/cola.linear_operators.rst
docs/package/cola.ops.rst
tests/test_decomps.py
tests/test_linalg.py
tests/test_operators.py
tests/test_pytree.py
tests/algorithms/test_arnoldi.py
tests/algorithms/test_cg.py
tests/algorithms/test_gmres.py
tests/algorithms/test_iram.py
tests/algorithms/test_lanczos.py
tests/algorithms/test_lobpcg.py
tests/algorithms/test_precond.py
tests/algorithms/test_qr.py
tests/algorithms/test_stochastic_lanczos_quad.py
tests/backends/test_backends.py
tests/data/1138_bus.mtx
tests/data/Tre20k.mtx
tests/data/cfd1.mtx
tests/data/finan512.mtx
tests/linalg/operator_market.py
tests/linalg/test_diagonal.py
tests/linalg/test_eig.py
tests/linalg/test_inverse.py
tests/linalg/test_logdet.py
tests/linalg/test_solve.py
tests/linalg/test_sqrt.py
tests/linalg/test_unary.py
tests/linalg/inverse/nonsquare_operator_market.py
tests/linalg/inverse/test_pinv.py
tests/linalg/svd/test_svd.py