.gitignore
.readthedocs.yaml
LICENSE
README.md
pyproject.toml
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/pull_request.yml
.github/workflows/push_to_main.yml
.github/workflows/run_test_suite.yml
cola/__init__.py
cola/annotations.py
cola/decompositions.py
cola/fns.py
cola/jax_fns.py
cola/np_fns.py
cola/torch_fns.py
cola/utils_test.py
cola/version.py
cola/algorithms/__init__.py
cola/algorithms/arnoldi.py
cola/algorithms/cg.py
cola/algorithms/diagonal_estimation.py
cola/algorithms/gmres.py
cola/algorithms/iram.py
cola/algorithms/lanczos.py
cola/algorithms/lobpcg.py
cola/algorithms/power_iteration.py
cola/algorithms/preconditioners.py
cola/algorithms/randomized_svd.py
cola/algorithms/slq.py
cola/algorithms/svrg.py
cola/linalg/__init__.py
cola/linalg/diag_trace.py
cola/linalg/eigs.py
cola/linalg/inv.py
cola/linalg/logdet.py
cola/linalg/nullspace.py
cola/linalg/pinv.py
cola/linalg/solve.py
cola/linalg/svd.py
cola/linalg/unary.py
cola/ops/__init__.py
cola/ops/operator_base.py
cola/ops/operators.py
cola/utils/__init__.py
cola/utils/control_flow.py
cola/utils/custom_autodiff.py
cola/utils/dispatch.py
cola/utils/jax_tqdm.py
cola/utils/torch_tqdm.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/testing.md
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/worksheet.ipynb
docs/package/cola.algorithms.rst
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_stochastic_lanczos_quad.py
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