f-dangel / curvlinops

91%

LAST BUILD ON BRANCH main

branch: SELECT

CHANGE BRANCH

x

Sync Branches

• No branch selected
• 3.0.1
• activation-hessian
• activation-hessian-linop
• add-ekfac-benchmark
• adjoint
• adjoint-jacobians
• adjoint-submatrix
• alt-ekfac
• always-full-tests
• arbitrary-inputs
• bcewithlogitsloss
• blockdiagonal-linear-operator
• bug-dtype-check-deterministic
• bug-kfac-device-change
• ci-update
• codex/replace-darglint-with-ruff-for-linting
• composition-rules
• copilot/add-fishertype-kfactype-enums
• curvature-linop
• deprecate-pkg-resources
• deprecate-python38
• development
• diagonal
• diagonal-linear-operator
• doc-eigenvalues
• docstring-linting-with-ruff
• double-gramian-contraction
• dtype-device-properties
• ef-linop
• eigh-linear-operator
• ekfac
• empirical-fisher
• empirical-gradient-computer
• empirical-risk-mixin
• eva
• example-benchmark
• example-fisher-mc
• example-trace
• execute-tests-in-float64
• expand-reduce
• extract-gradient-and-loss
• fail-first
• faster-ef
• faster-fisher-mc
• fisher-averaging
• fisher-linop
• fisher-mc
• fix-coveralls-badge
• fix-device
• fix-device-errors
• fix-ekfac-performance
• fix-enum
• fix-inverse-kfac
• fix-kfac-scale
• fix-markdown-syntax
• fix-mc-tests
• fix-mcfisher-scale
• fix-python-badge
• fix-scale
• fix-typo
• flake8-to-ruff
• foof
• frobenius-norm
• functional-estimators
• functional-ggnmp
• functorch-efmp
• functorch-fmp
• functorch-ggnmp
• functorch-hmp
• functorch-jacobians
• further-fix-ekfac-performance
• ggn-diagonal
• ggn-diagonal-pr
• ggn-linop
• hessian-blocks
• hessian-linop
• high-dim-outputs
• hutchinson
• hutchpp
• implement-rmatmul
• improve-benchmark-results
• improve-frobenius
• improve-spectral-density
• improve-tests
• improve-visual-tour
• io-collector
• io-collector-pr1
• io-collector-pr2
• issue-138-improve-links
• jacobian
• jacobian-linop
• jacobian-transposed
• joint-weight-and-bias
• kfac
• kfac-arbitrary-params-order
• kfac-as-blockdiag-kronecker
• kfac-canonicalization
• kfac-ce-loss
• kfac-computer
• kfac-conv2d
• kfac-damping
• kfac-ef
• kfac-ema
• kfac-inv
• kfac-linop
• kfac-properties-to-methods
• kfac-state-dict
• kfac-torch-only
• kfac-trace
• kfac-type-2
• kfac-with-inplace-activations
• kronecker-linear-operator
• kronecker-product
• lanczos-cache
• load-state-dict-model-func
• logo
• lsmr-inverse
• main
• matmat-list-shapes
• matvec-to-matmat
• merge-adjoint-and-is-vec-tests
• merge-linting-formatting
• modernize-type-annotations
• move-util-linops
• neumann-inverse
• neumann-inverse-example
• no-cpu
• only-bias-or-weight
• overwrite-model-warning
• papyan20traces-density
• parallelize-grad-output
• parallelize-sample-grad-output
• parallelize-tests
• partial-boundary-specification
• power-iteration-comparison
• pr1_deprecate_py
• pr2
• pr3
• prepare-3-0-1
• pydocstyle-coverage
• pytorch-cg
• pytorch-inverse
• pytorch-linop
• pytorch-spectral-density
• reduce-scipy
• ref-internal-mapping
• remove-caching
• remove-ekfac-activation-caching
• remove-kfac-inverse
• remove-kfac-side-effect
• remove-loss-average
• remove-to_device
• remover-iterator-conversion
• retry-kfac-inv
• rla-pytorch
• runame-patch-1
• source-button
• specify-num-data
• submatrices
• submatrix-example
• submatrix-pytorch
• support-division
• support-sequences-in-grad-output-sampler
• support-sequences-in-hessian-sqrt
• test-consecutive-matmats
• test-regression
• update-benchmark-a6000
• update-black
• update-changelog
• update-ci
• update-github-actions
• use-built-in-functorch
• use-pyproject-toml
• use-tuples-in-functionals
• v3-release-notes
• visual-tour
• weighted-ce-kfac
• xdiag
• xtrace

Build # 22080422973

Build Type

push

github

Committed by web-flow

Commit Message

[ADD] IO collector for KFAC (#253)

* [ADD] Collector for in/outputs of linear weight sharing layers

* [ADD] Collector for KFAC

* [DOC] Update changelog

* [REF] Improvements

---------

Co-authored-by: Felix Dangel <fdangel@vectorinstitute.ai>

Run Details

72 of 77 new or added lines in 2 files covered. (93.51%)

2211 of 2423 relevant lines covered (91.25%)

0.91 hits per line