.conda_env.yml
.envrc
.gitignore
.readthedocs.yaml
LICENSE
README.md
changelog.md
makefile
pyproject.toml
pytest.ini
setup.cfg
.github/workflows/lint-darglint.yaml
.github/workflows/lint-format.yaml
.github/workflows/lint-pydocstyle.yaml
.github/workflows/lint-ruff.yaml
.github/workflows/python-publish.yml
.github/workflows/test.yaml
curvlinops/__init__.py
curvlinops/_torch_base.py
curvlinops/ekfac.py
curvlinops/fisher.py
curvlinops/ggn.py
curvlinops/gradient_moments.py
curvlinops/hessian.py
curvlinops/inverse.py
curvlinops/jacobian.py
curvlinops/kfac.py
curvlinops/kfac_utils.py
curvlinops/sampling.py
curvlinops/submatrix.py
curvlinops/utils.py
curvlinops/diagonal/__init__.py
curvlinops/diagonal/epperly2024xtrace.py
curvlinops/diagonal/hutchinson.py
curvlinops/examples/__init__.py
curvlinops/examples/functorch.py
curvlinops/experimental/__init__.py
curvlinops/experimental/activation_hessian.py
curvlinops/norm/__init__.py
curvlinops/norm/hutchinson.py
curvlinops/papyan2020traces/__init__.py
curvlinops/papyan2020traces/spectrum.py
curvlinops/trace/__init__.py
curvlinops/trace/epperly2024xtrace.py
curvlinops/trace/hutchinson.py
curvlinops/trace/meyer2020hutch.py
curvlinops_for_pytorch.egg-info/PKG-INFO
curvlinops_for_pytorch.egg-info/SOURCES.txt
curvlinops_for_pytorch.egg-info/dependency_links.txt
curvlinops_for_pytorch.egg-info/requires.txt
curvlinops_for_pytorch.egg-info/top_level.txt
docs/examples/basic_usage/.gitignore
docs/examples/basic_usage/README.rst
docs/examples/basic_usage/benchmark_utils.py
docs/examples/basic_usage/curvature_matrices_log_abs.pdf
docs/examples/basic_usage/diagonal_estimation.pdf
docs/examples/basic_usage/example_benchmark.py
docs/examples/basic_usage/example_eigenvalues.py
docs/examples/basic_usage/example_fisher_monte_carlo.py
docs/examples/basic_usage/example_huggingface.py
docs/examples/basic_usage/example_inverses.py
docs/examples/basic_usage/example_matrix_vector_products.py
docs/examples/basic_usage/example_model_merging.py
docs/examples/basic_usage/example_submatrices.py
docs/examples/basic_usage/example_trace_diagonal_estimation.py
docs/examples/basic_usage/example_verification_spectral_density.py
docs/examples/basic_usage/example_visual_tour.py
docs/examples/basic_usage/memory_benchmark.py
docs/examples/basic_usage/toy_log_spectrum.pdf
docs/examples/basic_usage/toy_spectrum.pdf
docs/examples/basic_usage/trace_estimation.pdf
docs/examples/basic_usage/benchmark/peakmem_Empirical-Fisher_synthetic_imagenet_resnet50_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/peakmem_Empirical-Fisher_synthetic_imagenet_resnet50_cuda_matvec.json
docs/examples/basic_usage/benchmark/peakmem_Empirical-Fisher_synthetic_imagenet_resnet50_cuda_precompute.json
docs/examples/basic_usage/benchmark/peakmem_Empirical-Fisher_synthetic_shakespeare_nanogpt_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/peakmem_Empirical-Fisher_synthetic_shakespeare_nanogpt_cuda_matvec.json
docs/examples/basic_usage/benchmark/peakmem_Empirical-Fisher_synthetic_shakespeare_nanogpt_cuda_precompute.json
docs/examples/basic_usage/benchmark/peakmem_Generalized-Gauss-Newton_synthetic_imagenet_resnet50_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/peakmem_Generalized-Gauss-Newton_synthetic_imagenet_resnet50_cuda_matvec.json
docs/examples/basic_usage/benchmark/peakmem_Generalized-Gauss-Newton_synthetic_imagenet_resnet50_cuda_precompute.json
docs/examples/basic_usage/benchmark/peakmem_Generalized-Gauss-Newton_synthetic_shakespeare_nanogpt_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/peakmem_Generalized-Gauss-Newton_synthetic_shakespeare_nanogpt_cuda_matvec.json
docs/examples/basic_usage/benchmark/peakmem_Generalized-Gauss-Newton_synthetic_shakespeare_nanogpt_cuda_precompute.json
docs/examples/basic_usage/benchmark/peakmem_Hessian_synthetic_imagenet_resnet50_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/peakmem_Hessian_synthetic_imagenet_resnet50_cuda_matvec.json
docs/examples/basic_usage/benchmark/peakmem_Hessian_synthetic_imagenet_resnet50_cuda_precompute.json
docs/examples/basic_usage/benchmark/peakmem_Hessian_synthetic_shakespeare_nanogpt_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/peakmem_Hessian_synthetic_shakespeare_nanogpt_cuda_matvec.json
docs/examples/basic_usage/benchmark/peakmem_Hessian_synthetic_shakespeare_nanogpt_cuda_precompute.json
docs/examples/basic_usage/benchmark/peakmem_KFAC-inverse_synthetic_imagenet_resnet50_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/peakmem_KFAC-inverse_synthetic_imagenet_resnet50_cuda_matvec.json
docs/examples/basic_usage/benchmark/peakmem_KFAC-inverse_synthetic_imagenet_resnet50_cuda_precompute.json
docs/examples/basic_usage/benchmark/peakmem_KFAC-inverse_synthetic_shakespeare_nanogpt_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/peakmem_KFAC-inverse_synthetic_shakespeare_nanogpt_cuda_matvec.json
docs/examples/basic_usage/benchmark/peakmem_KFAC-inverse_synthetic_shakespeare_nanogpt_cuda_precompute.json
docs/examples/basic_usage/benchmark/peakmem_KFAC_synthetic_imagenet_resnet50_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/peakmem_KFAC_synthetic_imagenet_resnet50_cuda_matvec.json
docs/examples/basic_usage/benchmark/peakmem_KFAC_synthetic_imagenet_resnet50_cuda_precompute.json
docs/examples/basic_usage/benchmark/peakmem_KFAC_synthetic_shakespeare_nanogpt_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/peakmem_KFAC_synthetic_shakespeare_nanogpt_cuda_matvec.json
docs/examples/basic_usage/benchmark/peakmem_KFAC_synthetic_shakespeare_nanogpt_cuda_precompute.json
docs/examples/basic_usage/benchmark/peakmem_Monte-Carlo-Fisher_synthetic_imagenet_resnet50_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/peakmem_Monte-Carlo-Fisher_synthetic_imagenet_resnet50_cuda_matvec.json
docs/examples/basic_usage/benchmark/peakmem_Monte-Carlo-Fisher_synthetic_imagenet_resnet50_cuda_precompute.json
docs/examples/basic_usage/benchmark/peakmem_Monte-Carlo-Fisher_synthetic_shakespeare_nanogpt_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/peakmem_Monte-Carlo-Fisher_synthetic_shakespeare_nanogpt_cuda_matvec.json
docs/examples/basic_usage/benchmark/peakmem_Monte-Carlo-Fisher_synthetic_shakespeare_nanogpt_cuda_precompute.json
docs/examples/basic_usage/benchmark/peakmem_synthetic_imagenet_resnet50_cuda.pdf
docs/examples/basic_usage/benchmark/peakmem_synthetic_shakespeare_nanogpt_cuda.pdf
docs/examples/basic_usage/benchmark/time_Empirical-Fisher_synthetic_imagenet_resnet50_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/time_Empirical-Fisher_synthetic_imagenet_resnet50_cuda_matvec.json
docs/examples/basic_usage/benchmark/time_Empirical-Fisher_synthetic_imagenet_resnet50_cuda_precompute.json
docs/examples/basic_usage/benchmark/time_Empirical-Fisher_synthetic_shakespeare_nanogpt_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/time_Empirical-Fisher_synthetic_shakespeare_nanogpt_cuda_matvec.json
docs/examples/basic_usage/benchmark/time_Empirical-Fisher_synthetic_shakespeare_nanogpt_cuda_precompute.json
docs/examples/basic_usage/benchmark/time_Generalized-Gauss-Newton_synthetic_imagenet_resnet50_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/time_Generalized-Gauss-Newton_synthetic_imagenet_resnet50_cuda_matvec.json
docs/examples/basic_usage/benchmark/time_Generalized-Gauss-Newton_synthetic_imagenet_resnet50_cuda_precompute.json
docs/examples/basic_usage/benchmark/time_Generalized-Gauss-Newton_synthetic_shakespeare_nanogpt_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/time_Generalized-Gauss-Newton_synthetic_shakespeare_nanogpt_cuda_matvec.json
docs/examples/basic_usage/benchmark/time_Generalized-Gauss-Newton_synthetic_shakespeare_nanogpt_cuda_precompute.json
docs/examples/basic_usage/benchmark/time_Hessian_synthetic_imagenet_resnet50_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/time_Hessian_synthetic_imagenet_resnet50_cuda_matvec.json
docs/examples/basic_usage/benchmark/time_Hessian_synthetic_imagenet_resnet50_cuda_precompute.json
docs/examples/basic_usage/benchmark/time_Hessian_synthetic_shakespeare_nanogpt_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/time_Hessian_synthetic_shakespeare_nanogpt_cuda_matvec.json
docs/examples/basic_usage/benchmark/time_Hessian_synthetic_shakespeare_nanogpt_cuda_precompute.json
docs/examples/basic_usage/benchmark/time_KFAC-inverse_synthetic_imagenet_resnet50_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/time_KFAC-inverse_synthetic_imagenet_resnet50_cuda_matvec.json
docs/examples/basic_usage/benchmark/time_KFAC-inverse_synthetic_imagenet_resnet50_cuda_precompute.json
docs/examples/basic_usage/benchmark/time_KFAC-inverse_synthetic_shakespeare_nanogpt_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/time_KFAC-inverse_synthetic_shakespeare_nanogpt_cuda_matvec.json
docs/examples/basic_usage/benchmark/time_KFAC-inverse_synthetic_shakespeare_nanogpt_cuda_precompute.json
docs/examples/basic_usage/benchmark/time_KFAC_synthetic_imagenet_resnet50_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/time_KFAC_synthetic_imagenet_resnet50_cuda_matvec.json
docs/examples/basic_usage/benchmark/time_KFAC_synthetic_imagenet_resnet50_cuda_precompute.json
docs/examples/basic_usage/benchmark/time_KFAC_synthetic_shakespeare_nanogpt_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/time_KFAC_synthetic_shakespeare_nanogpt_cuda_matvec.json
docs/examples/basic_usage/benchmark/time_KFAC_synthetic_shakespeare_nanogpt_cuda_precompute.json
docs/examples/basic_usage/benchmark/time_Monte-Carlo-Fisher_synthetic_imagenet_resnet50_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/time_Monte-Carlo-Fisher_synthetic_imagenet_resnet50_cuda_matvec.json
docs/examples/basic_usage/benchmark/time_Monte-Carlo-Fisher_synthetic_imagenet_resnet50_cuda_precompute.json
docs/examples/basic_usage/benchmark/time_Monte-Carlo-Fisher_synthetic_shakespeare_nanogpt_cuda_gradient_and_loss.json
docs/examples/basic_usage/benchmark/time_Monte-Carlo-Fisher_synthetic_shakespeare_nanogpt_cuda_matvec.json
docs/examples/basic_usage/benchmark/time_Monte-Carlo-Fisher_synthetic_shakespeare_nanogpt_cuda_precompute.json
docs/examples/basic_usage/benchmark/time_synthetic_imagenet_resnet50_cuda.pdf
docs/examples/basic_usage/benchmark/time_synthetic_shakespeare_nanogpt_cuda.pdf
docs/rtd/.gitignore
docs/rtd/Makefile
docs/rtd/conf.py
docs/rtd/index.rst
docs/rtd/internals.rst
docs/rtd/linops.rst
docs/rtd/make.bat
docs/rtd/usage.rst
docs/rtd/assets/.gitignore
docs/rtd/assets/convert.sh
docs/rtd/assets/hessian.pdf
docs/rtd/assets/logo.pdf
docs/rtd/assets/logo.png
docs/rtd/assets/logo.svg
docs/rtd/assets/logo.tex
docs/rtd/assets/torch.pdf
test/__init__.py
test/cases.py
test/conftest.py
test/kfac_cases.py
test/test__torch_base.py
test/test_ekfac.py
test/test_fisher.py
test/test_ggn.py
test/test_gradient_moments.py
test/test_hessian.py
test/test_inverse.py
test/test_jacobian.py
test/test_kfac.py
test/test_submatrix.py
test/test_utils.py
test/utils.py
test/diagonal/__init__.py
test/diagonal/test_epperly2024xtrace.py
test/diagonal/test_hutchinson.py
test/experimental/__init__.py
test/experimental/test_activation_hessian.py
test/norm/__init__.py
test/norm/test_hutchinson.py
test/papyan2020traces/__init__.py
test/papyan2020traces/test_spectrum.py
test/trace/__init__.py
test/trace/test_epperly2024xtrace.py
test/trace/test_hutchinson.py
test/trace/test_meyer2020hutch.py